#include <bits/stdc++.h> using namespace std; int n, k, res, t[2010]; int main() { cin >> n >> k; for (int i=0; i<n; i++) { cin >> t[i]; } sort(t, t+n, greater<int>()); res = k; for (int i=k; i<n; i++) { if (t[i] == t[k-1]) res++; } cout << res << endl; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | #include <bits/stdc++.h> using namespace std; int n, k, res, t[2010]; int main() { cin >> n >> k; for (int i=0; i<n; i++) { cin >> t[i]; } sort(t, t+n, greater<int>()); res = k; for (int i=k; i<n; i++) { if (t[i] == t[k-1]) res++; } cout << res << endl; } |