Submission #13,253

2025-01-07 10:56:55

cpp17

Accepted 50/50

18ms

512 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++;
		}
	}	
	
	//m=((z-k)*2)+1;
	if(z<k) {
		cout<<0;
		return 0;
	}
	long long m=0;
	for(int i=k; i<=z;i++)
		{
			m+=z-i+1;
		}
	cout<<m;
}

	Sum	Test	Verdict		Time	Memory
base	50/50
		1	Accepted	0/0	1ms	316 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	316 KiB
		12	Accepted	2/2	1ms	316 KiB
		13	Accepted	3/3	1ms	316 KiB
		14	Accepted	2/2	17ms	396 KiB
		15	Accepted	2/2	17ms	400 KiB
		16	Accepted	3/3	17ms	512 KiB
		17	Accepted	3/3	17ms	392 KiB
		18	Accepted	3/3	17ms	396 KiB
		19	Accepted	3/3	17ms	316 KiB
		20	Accepted	3/3	17ms	316 KiB
		21	Accepted	3/3	18ms	512 KiB
		22	Accepted	3/3	17ms	400 KiB