Submission #14,409

2025-01-10 18:13:23

cpp17

Accepted 50/50

18ms

808 KiB

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int n, k;
    long db = 0;
    cin >> n >> k;
    vector<int> v(n);
    for (int i = 0; i < n; i ++) {
        cin >> v[i];
        if (v[i] == 1) db += 1;
    }
    if (db < k) cout << 0;
    else cout << (db + 1 - k)*(db - k + 2) / 2;
    return 0;
}

	Sum	Test	Verdict		Time	Memory
base	50/50
		1	Accepted	0/0	1ms	316 KiB
		2	Accepted	0/0	1ms	512 KiB
		3	Accepted	0/0	18ms	564 KiB
		4	Accepted	2/2	1ms	316 KiB
		5	Accepted	2/2	1ms	316 KiB
		6	Accepted	2/2	1ms	508 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	356 KiB
		11	Accepted	3/3	1ms	316 KiB
		12	Accepted	2/2	1ms	316 KiB
		13	Accepted	3/3	1ms	552 KiB
		14	Accepted	2/2	17ms	776 KiB
		15	Accepted	2/2	17ms	788 KiB
		16	Accepted	3/3	18ms	564 KiB
		17	Accepted	3/3	17ms	784 KiB
		18	Accepted	3/3	18ms	808 KiB
		19	Accepted	3/3	18ms	564 KiB
		20	Accepted	3/3	18ms	564 KiB
		21	Accepted	3/3	18ms	568 KiB
		22	Accepted	3/3	18ms	564 KiB