#include <iostream> #include <algorithm> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, a = 0, pom; cin >> n >> k; if(n!=1) { if(k!=n) { int kos[n]; for(int i=0; i<=n-1; i++) { cin >> kos[i]; } sort(kos, kos+n); pom=kos[n-k]; for(int i=n-k-1; i>=0; i--) { if(kos[i]==pom) a++; else break; } a+=k; } else { a = k; } } else { a = 1; } cout << a; return 0; }
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | #include <iostream> #include <algorithm> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, a = 0, pom; cin >> n >> k; if(n!=1) { if(k!=n) { int kos[n]; for(int i=0; i<=n-1; i++) { cin >> kos[i]; } sort(kos, kos+n); pom=kos[n-k]; for(int i=n-k-1; i>=0; i--) { if(kos[i]==pom) a++; else break; } a+=k; } else { a = k; } } else { a = 1; } cout << a; return 0; } |