257662026-03-01 20:37:50horvayzsomborVirágos rét (50 pont)pypy3Runtime error 0/5072ms22840 KiB
#include <iostream>
#include <vector>

using namespace std;

int main()
{
    int n, k;
    cin >> n >> k;

    vector<char> v(n);

    for(int i = 0; i < n; i++)
    {
        cin >> v[i];
    }

    vector<int> s(n + 1);

    for(int i = n - 1; i >= 0; i--)
    {
        s[i] = s[i + 1];   
        if(v[i] == '1') s[i]++; 
    }

    int ans = 0;

    int cnt = 0;
    int j = 0;

    for(int i = 0; i < n; i++)
    {
        if(v[i] == '1')
        {
            while(j < n && cnt != k)
            {
                if(v[j] == '1') cnt++;
                j++;
            }

            if(cnt == k) ans += s[j - 1];

            cnt--;
        }
    }

    cout << ans;

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base0/50
1Runtime error0/071ms22680 KiB
2Runtime error0/064ms22588 KiB
3Runtime error0/071ms22740 KiB
4Runtime error0/264ms22756 KiB
5Runtime error0/265ms22784 KiB
6Runtime error0/272ms22756 KiB
7Runtime error0/364ms22608 KiB
8Runtime error0/371ms22756 KiB
9Runtime error0/372ms22708 KiB
10Runtime error0/264ms22616 KiB
11Runtime error0/371ms22692 KiB
12Runtime error0/264ms22656 KiB
13Runtime error0/372ms22756 KiB
14Runtime error0/264ms22732 KiB
15Runtime error0/264ms22816 KiB
16Runtime error0/372ms22760 KiB
17Runtime error0/372ms22624 KiB
18Runtime error0/365ms22756 KiB
19Runtime error0/364ms22760 KiB
20Runtime error0/371ms22784 KiB
21Runtime error0/370ms22752 KiB
22Runtime error0/364ms22840 KiB