#include <bits/stdc++.h> using namespace std; #define TESTS 0 const int N = 2100; int n, k, a[N]; void solve() { cin >> n >> k; for (int i = 1; i <= n; i++) cin >> a[i]; sort(a+1, a+n+1); reverse(a+1, a+n+1); int cnt = k; for (int i = k+1; i <= n && a[i] == a[k]; i++) ++cnt; cout << cnt << "\n"; } int main() { int tt = 1; #if TESTS cin >> tt; #endif while(tt--) { 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 26 27 28 29 30 31 32 | #include <bits/stdc++.h> using namespace std; #define TESTS 0 const int N = 2100; int n, k, a[N]; void solve() { cin >> n >> k; for (int i = 1; i <= n; i++) cin >> a[i]; sort(a+1, a+n+1); reverse(a+1, a+n+1); int cnt = k; for (int i = k+1; i <= n && a[i] == a[k]; i++) ++cnt; cout << cnt << "\n"; } int main() { int tt = 1; #if TESTS cin >> tt; #endif while(tt--) { solve(); } } |