171652025-05-26 21:41:28MolnárÚthasználati díj (75 pont)cpp17Elfogadva 75/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;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base75/75
1Elfogadva0/01ms316 KiB
2Elfogadva0/01ms316 KiB
3Elfogadva0/017ms408 KiB
4Elfogadva5/51ms328 KiB
5Elfogadva5/51ms316 KiB
6Elfogadva5/51ms316 KiB
7Elfogadva5/51ms316 KiB
8Elfogadva5/51ms316 KiB
9Elfogadva5/51ms316 KiB
10Elfogadva5/51ms316 KiB
11Elfogadva5/51ms316 KiB
12Elfogadva5/51ms508 KiB
13Elfogadva5/51ms316 KiB
14Elfogadva5/518ms396 KiB
15Elfogadva5/517ms396 KiB
16Elfogadva5/517ms400 KiB
17Elfogadva5/517ms400 KiB
18Elfogadva5/517ms400 KiB