Submission #21,462

2026-01-13 09:24:54

cpp17

Accepted 50/50

8ms

1124 KiB

#include <iostream>
using namespace std;

int main()
{   ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    long long n, k, p=0;
    cin >> n >> k;
    long long t[n];
    for(int i=0; i<n; i++)
    {
        cin >> t[i];
        if(t[i]==1)
        {
            p++;
        }

    }
    if(k>p)
    {
        cout << 0;
        return 0;
    }
    cout << (p+1-k)*(p+2-k)/2;



    return 0;
}

	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	8ms	1084 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	7ms	1104 KiB
		15	Accepted	2/2	8ms	936 KiB
		16	Accepted	3/3	8ms	1076 KiB
		17	Accepted	3/3	8ms	1076 KiB
		18	Accepted	3/3	8ms	1076 KiB
		19	Accepted	3/3	8ms	1064 KiB
		20	Accepted	3/3	8ms	1076 KiB
		21	Accepted	3/3	8ms	1124 KiB
		22	Accepted	3/3	8ms	1096 KiB