Submission #13,233

2025-01-07 09:46:34

cpp17

Accepted 50/50

8ms

508 KiB

#include <bits/stdc++.h>
using namespace std;
#define forL(len) for (ll i = 1; i <= len; i++)
#define ll long long int

int main() {
    cin.tie(0)->sync_with_stdio(0);
    ll n,k,a=0,tobb=0; cin >> n >> k;
    int be;
    forL(n)
    {
        cin >> be;
        if (be == 1) a++;
    }
    
    for (ll i = 1; i <= a-k+1; i++)
    {
        tobb += i;
    }
    cout << tobb << endl;
    return 0;
}

	Sum	Test	Verdict		Time	Memory
base	50/50
		1	Accepted	0/0	1ms	316 KiB
		2	Accepted	0/0	1ms	508 KiB
		3	Accepted	0/0	8ms	316 KiB
		4	Accepted	2/2	1ms	316 KiB
		5	Accepted	2/2	1ms	316 KiB
		6	Accepted	2/2	1ms	316 KiB
		7	Accepted	3/3	1ms	316 KiB
		8	Accepted	3/3	1ms	508 KiB
		9	Accepted	3/3	1ms	328 KiB
		10	Accepted	2/2	1ms	508 KiB
		11	Accepted	3/3	1ms	316 KiB
		12	Accepted	2/2	1ms	316 KiB
		13	Accepted	3/3	1ms	316 KiB
		14	Accepted	2/2	7ms	424 KiB
		15	Accepted	2/2	7ms	424 KiB
		16	Accepted	3/3	8ms	424 KiB
		17	Accepted	3/3	8ms	316 KiB
		18	Accepted	3/3	8ms	424 KiB
		19	Accepted	3/3	8ms	428 KiB
		20	Accepted	3/3	8ms	420 KiB
		21	Accepted	3/3	8ms	420 KiB
		22	Accepted	3/3	8ms	424 KiB