#include <bits/stdc++.h> using namespace std; int main() { int n, k; cin >> n >> k; vector<int> v(n); for(int i=0; i<n; ++i) cin >> v[i]; sort(v.begin(), v.end()); reverse(v.begin(), v.end()); int j = k; while(j<n && v[j]==v[k-1]) { ++j; } cout << j << endl; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | #include <bits/stdc++.h> using namespace std; int main() { int n, k; cin >> n >> k; vector<int> v(n); for(int i=0; i<n; ++i) cin >> v[i]; sort(v.begin(), v.end()); reverse(v.begin(), v.end()); int j = k; while(j<n && v[j]==v[k-1]) { ++j; } cout << j << endl; } |