131632025-01-06 19:41:09bzyÚthasználati díj (75 pont)cpp17Accepted 75/7518ms884 KiB
#include <bits/stdc++.h>
using namespace std;
#define ll  long long

int main()
{
    int n, k; cin >> n >> k;
    vector <int> A(n);
    int utidijas = 0;
    for(int i = 0; i < n; i++)
    {
        cin >> A[i];
        if(A[i] == 1) utidijas++;
    }
    int ans = 0;
    if(utidijas < k)
    {
        ans += utidijas * (utidijas-1) / 2;
    }
    else
    {
        ans += (utidijas - k + 1) * (k-1);
        ans += (k-1) * (k-2) / 2;
    }
    cout << ans;
    return 0;
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/01ms316 KiB
2Accepted0/01ms544 KiB
3Accepted0/017ms564 KiB
4Accepted5/51ms316 KiB
5Accepted5/51ms316 KiB
6Accepted5/51ms316 KiB
7Accepted5/51ms500 KiB
8Accepted5/51ms316 KiB
9Accepted5/51ms316 KiB
10Accepted5/51ms316 KiB
11Accepted5/51ms316 KiB
12Accepted5/51ms508 KiB
13Accepted5/51ms508 KiB
14Accepted5/518ms884 KiB
15Accepted5/517ms564 KiB
16Accepted5/517ms804 KiB
17Accepted5/517ms796 KiB
18Accepted5/517ms564 KiB