114122024-09-13 09:55:23madvirVirágos rét (50 pont)cpp17Elfogadva 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;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base50/50
1Elfogadva0/02ms360 KiB
2Elfogadva0/03ms360 KiB
3Elfogadva0/018ms384 KiB
4Elfogadva2/23ms276 KiB
5Elfogadva2/23ms364 KiB
6Elfogadva2/23ms384 KiB
7Elfogadva3/32ms384 KiB
8Elfogadva3/32ms256 KiB
9Elfogadva3/33ms488 KiB
10Elfogadva2/23ms420 KiB
11Elfogadva3/33ms360 KiB
12Elfogadva2/23ms256 KiB
13Elfogadva3/33ms360 KiB
14Elfogadva2/217ms512 KiB
15Elfogadva2/217ms384 KiB
16Elfogadva3/318ms256 KiB
17Elfogadva3/317ms384 KiB
18Elfogadva3/318ms360 KiB
19Elfogadva3/318ms376 KiB
20Elfogadva3/318ms504 KiB
21Elfogadva3/318ms504 KiB
22Elfogadva3/318ms384 KiB