131632025-01-06 19:41:09bzyÚthasználati díj (75 pont)cpp17Elfogadva 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;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base75/75
1Elfogadva0/01ms316 KiB
2Elfogadva0/01ms544 KiB
3Elfogadva0/017ms564 KiB
4Elfogadva5/51ms316 KiB
5Elfogadva5/51ms316 KiB
6Elfogadva5/51ms316 KiB
7Elfogadva5/51ms500 KiB
8Elfogadva5/51ms316 KiB
9Elfogadva5/51ms316 KiB
10Elfogadva5/51ms316 KiB
11Elfogadva5/51ms316 KiB
12Elfogadva5/51ms508 KiB
13Elfogadva5/51ms508 KiB
14Elfogadva5/518ms884 KiB
15Elfogadva5/517ms564 KiB
16Elfogadva5/517ms804 KiB
17Elfogadva5/517ms796 KiB
18Elfogadva5/517ms564 KiB