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