171612025-05-26 21:36:47MolnárÚthasználati díj (75 pont)cpp17Wrong answer 60/7517ms508 KiB
#include <iostream>
using namespace std;

int main()
{
    int n, k, x;
    cin >> n >> k;
    int j = 0;

    for (int i = 0; i < n; i++)
    {
        cin >> x;

        if (x == 1)
        {
            j++;
        }
    }

    int ossz = 0;

    // for (int i = 0; i < j; i++)
    // {
    //     // ossz += min(k - 1, j - i - 1);
    //     if (k - 1 < j - i - 1)
    //         ossz = ossz + k - 1;
    //     else
    //         ossz = ossz + j - i - 1;
    // }

    if (k - 1 <= j) {
        ossz = (j - k + 1) * (k - 1) + (k - 1) * (k - 2) / 2;
    }

    cout << ossz;

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base60/75
1Accepted0/01ms508 KiB
2Accepted0/01ms316 KiB
3Wrong answer0/017ms316 KiB
4Accepted5/51ms316 KiB
5Accepted5/51ms316 KiB
6Wrong answer0/51ms316 KiB
7Accepted5/51ms316 KiB
8Accepted5/51ms508 KiB
9Accepted5/51ms316 KiB
10Accepted5/51ms316 KiB
11Accepted5/51ms508 KiB
12Accepted5/51ms316 KiB
13Wrong answer0/51ms316 KiB
14Accepted5/517ms396 KiB
15Accepted5/517ms316 KiB
16Accepted5/517ms400 KiB
17Accepted5/517ms396 KiB
18Wrong answer0/517ms316 KiB