203352026-01-06 14:44:32algoproÚthasználati díj (75 pont)cpp17Accepted 75/7519ms940 KiB
// UUID: f9c6aabc-4905-4a98-9f2e-5cadaf514b16
#include <bits/stdc++.h>
#include <vector>
using namespace std;

/*
10 3
0 1 1 0 0 0 1 1 0 1
  2 2       2 1   0
*/

int main() {
  int n,k,sz=0,f=0;
  cin>>n>>k;
  vector<int>v(n);
  for(int i=0;i<n;i++){
    cin>>v[i];
    if (v[i]==1) {
      f++;
    }
  }
  k--;
  for(int i=0;i<n;i++){
    if (v[i]==1) {
      f--;
      if(k>=f){
        sz+=f;
      }else{
        sz+=k;
      }
    }
  }
  cout<<sz;
}
SubtaskSumTestVerdictTimeMemory
base75/75
1Accepted0/01ms316 KiB
2Accepted0/01ms508 KiB
3Accepted0/018ms564 KiB
4Accepted5/51ms500 KiB
5Accepted5/51ms316 KiB
6Accepted5/51ms316 KiB
7Accepted5/51ms804 KiB
8Accepted5/51ms316 KiB
9Accepted5/51ms316 KiB
10Accepted5/51ms316 KiB
11Accepted5/51ms508 KiB
12Accepted5/51ms316 KiB
13Accepted5/52ms500 KiB
14Accepted5/519ms820 KiB
15Accepted5/519ms888 KiB
16Accepted5/519ms812 KiB
17Accepted5/519ms940 KiB
18Accepted5/519ms820 KiB