Submission #20,769

20769

2026-01-08 19:45:56

cpp17

Wrong answer 42/50

19ms

808 KiB

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    int N, K;
    cin >> N >> K;
    int db1 = 0;
    vector <int> R(N+1);
    for (int i = 1; i <= N; i++) {
        cin >> R[i];
        if (R[i]==1) { db1++; }
    }

    if (db1 < K) { cout << 0; }else if (db1 == K) { cout << 1; }else {
        int ossz = 0;
        for (int i = 1; i <= N; i++) {
            if (R[i]==1) {
                if (db1 >= K) ossz += db1-K+1;
                db1--;
            }
        }
        cout << ossz;
    }
    return 0;
}

	Sum	Test	Verdict		Time	Memory
base	42/50
		1	Accepted	0/0	1ms	316 KiB
		2	Accepted	0/0	2ms	316 KiB
		3	Accepted	0/0	19ms	756 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	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	780 KiB
		15	Wrong answer	0/2	17ms	564 KiB
		16	Accepted	3/3	19ms	564 KiB
		17	Accepted	3/3	17ms	788 KiB
		18	Wrong answer	0/3	19ms	564 KiB
		19	Accepted	3/3	18ms	564 KiB
		20	Accepted	3/3	18ms	808 KiB
		21	Accepted	3/3	19ms	804 KiB
		22	Wrong answer	0/3	18ms	564 KiB