114122024-09-13 09:55:23madvirVirágos rét (50 pont)cpp17Accepted 50/5018ms512 KiB
#include <iostream>
#include <set>

using namespace std;

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

    ///1. megoldas
    /*for(int i=0; i<n; i++) {
        cin >> ret[i];
        if(ret[i]==1) {
            vir[s]=i;
            s++;
        }
    }

    int k0=0, st=0, fin=1, card=0;

    while(st<s-k) {
        while(k0<k && fin!=s-1) {
            fin++;
        }
        if(k0!=k) {
            cout << 0;
            st=s;
        }
        else {
            card++;
            card+=s-1-fin;
        }
    }

    cout << card;*/

    long long x, ret=0, card=0;

    for(int i=0; i<n; i++) {
        cin >> x;
        if(x==1) ret++;
    }

    if(ret<k) card=0;
    else {
        if(ret==k) card=1;
        else {
            int ki=0;
            for(int i=0; i<ret-k+1; i++) {
                card+=ret-k+1-ki;
                ki++;
            }
        }
    }

    cout<< card;

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/02ms360 KiB
2Accepted0/03ms360 KiB
3Accepted0/018ms384 KiB
4Accepted2/23ms276 KiB
5Accepted2/23ms364 KiB
6Accepted2/23ms384 KiB
7Accepted3/32ms384 KiB
8Accepted3/32ms256 KiB
9Accepted3/33ms488 KiB
10Accepted2/23ms420 KiB
11Accepted3/33ms360 KiB
12Accepted2/23ms256 KiB
13Accepted3/33ms360 KiB
14Accepted2/217ms512 KiB
15Accepted2/217ms384 KiB
16Accepted3/318ms256 KiB
17Accepted3/317ms384 KiB
18Accepted3/318ms360 KiB
19Accepted3/318ms376 KiB
20Accepted3/318ms504 KiB
21Accepted3/318ms504 KiB
22Accepted3/318ms384 KiB