#include<bits/stdc++.h> using namespace std; void solve() { int k, n; cin >> n >> k; vector<int> scores(n, 0); for (int i = 0; i < n; i++) { cin >> scores[i]; } sort(scores.begin(), scores.end()); for (int i = k; i <= n; i++) { if (scores[n-i] != scores[n-k]){ cout << i-1 << "\n"; return; } } cout << n << "\n"; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); solve(); }
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 | #include<bits/stdc++.h> using namespace std; void solve() { int k, n; cin >> n >> k; vector<int> scores(n, 0); for (int i = 0; i < n; i++) { cin >> scores[i]; } sort(scores.begin(), scores.end()); for (int i = k; i <= n; i++) { if (scores[n-i] != scores[n-k]){ cout << i-1 << "\n"; return; } } cout << n << "\n"; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); solve(); } |