171632025-05-26 21:38:49MolnárÚthasználati díj (75 pont)cpp17Wrong answer 55/7517ms532 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 (j - k + 1 > 0) ossz += (j - k + 1) * (k - 1);
    ossz += (k - 1) * (k - 2) / 2;

    cout << ossz;

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