171642025-05-26 21:39:56MolnárÚthasználati díj (75 pont)cpp17Wrong answer 0/7518ms508 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);
    if (k - 1 > j) ossz += (k - 1) * (k - 2) / 2;
    else ossz += j * (j - 1) / 2;

    cout << ossz;

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