Submission #13,239

2025-01-07 10:22:59

cpp17

Accepted 50/50

17ms

508 KiB

// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

int main() {
	int n,k;
	cin>>n>>k;
	int sz=0;
	int z=0;
	for(int i=0; i<n;i++){
		cin>>sz;
		if (sz==1) {
			z++;
		}
	}	
	

	if(z<k) {
		cout<<0;
		return 0;
	}
	long long m=0;
	 while (z >= k) {
        m += z - k + 1;
        z--;
    }
	
	
	cout<<m;
}

	Sum	Test	Verdict		Time	Memory
base	50/50
		1	Accepted	0/0	1ms	508 KiB
		2	Accepted	0/0	1ms	316 KiB
		3	Accepted	0/0	17ms	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	316 KiB
		9	Accepted	3/3	1ms	316 KiB
		10	Accepted	2/2	1ms	316 KiB
		11	Accepted	3/3	1ms	332 KiB
		12	Accepted	2/2	1ms	316 KiB
		13	Accepted	3/3	1ms	496 KiB
		14	Accepted	2/2	17ms	400 KiB
		15	Accepted	2/2	17ms	400 KiB
		16	Accepted	3/3	17ms	400 KiB
		17	Accepted	3/3	17ms	400 KiB
		18	Accepted	3/3	17ms	508 KiB
		19	Accepted	3/3	17ms	500 KiB
		20	Accepted	3/3	17ms	400 KiB
		21	Accepted	3/3	17ms	400 KiB
		22	Accepted	3/3	17ms	316 KiB