228472026-01-15 21:12:39helloworldVirágos rét (50 pont)cpp17Accepted 50/5017ms772 KiB
#include <stdlib.h>
#include <iostream>
using namespace std;

int N=0;
int M=0;
int a=0;
int db=0;




int main()
{
    cin>>N>>M;
    int szabad[N+1];
    for(int i=1; i<=N; i++){
        cin>>a;
        if(a==1){
            db++;
            szabad[db]=i;
        }
    }

   // if(db<M)cout<<0<<" ";

    /*for(int i=1; i<=db; i++){
        cout<<szabad[i]<<" ";
    }
    cout<<"\n";*/

    /*else{
        for(int pos=1; (pos+M-1)<=db; pos++){
            int cur=M;
            while(pos+cur-1<=db){
                //cout<<szabad[pos]<<"-->"<<szabad[pos+cur-1]<<" ";
                cur++;
                lehetseges++;
            }
        }
        cout<<lehetseges;
    }*/


    //ciklusok nelkul
    if(db<M)cout<<0;
    else {
        long long k = db - M + 1;
        long long lehetseges = k * (k + 1) / 2;
        cout << lehetseges;
    }

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms316 KiB
2Accepted0/01ms316 KiB
3Accepted0/017ms316 KiB
4Accepted2/21ms316 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms316 KiB
7Accepted3/31ms400 KiB
8Accepted3/31ms388 KiB
9Accepted3/31ms316 KiB
10Accepted2/21ms500 KiB
11Accepted3/31ms356 KiB
12Accepted2/21ms316 KiB
13Accepted3/31ms316 KiB
14Accepted2/217ms772 KiB
15Accepted2/217ms768 KiB
16Accepted3/317ms548 KiB
17Accepted3/317ms316 KiB
18Accepted3/317ms584 KiB
19Accepted3/317ms544 KiB
20Accepted3/317ms564 KiB
21Accepted3/317ms468 KiB
22Accepted3/317ms664 KiB