#include <iostream> #include <bits/stdc++.h> using namespace std; int n, k, a,wynik,maks; vector <int> t; int main() { cin>>n>>k; for(int i=0; i<n; i++) { cin>>a; t.push_back(a); } if(k>=n) { cout<<n; } else{ sort(t.begin(), t.end()); maks=t[n-k]; for(int i=n-k-1; i>=0; i--) { if(maks==t[i]) { wynik++; } } cout<<wynik+k;} return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | #include <iostream> #include <bits/stdc++.h> using namespace std; int n, k, a,wynik,maks; vector <int> t; int main() { cin>>n>>k; for(int i=0; i<n; i++) { cin>>a; t.push_back(a); } if(k>=n) { cout<<n; } else{ sort(t.begin(), t.end()); maks=t[n-k]; for(int i=n-k-1; i>=0; i--) { if(maks==t[i]) { wynik++; } } cout<<wynik+k;} return 0; } |