#include <iostream> #include <algorithm> using namespace std; int n, k, a, t[2005], zl[2005], wyn; int main() { cin >> n >> k; for(int i=0; i<n; ++i) { cin >> a; t[i]=a; zl[t[i]]++; } sort(t, t+n); for(int i=n-1; i>=0; --i) { k--; zl[t[i]]--; wyn++; if(k==0) { wyn=wyn+zl[t[i]]; cout << wyn; return 0; } } cout << wyn; 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 | #include <iostream> #include <algorithm> using namespace std; int n, k, a, t[2005], zl[2005], wyn; int main() { cin >> n >> k; for(int i=0; i<n; ++i) { cin >> a; t[i]=a; zl[t[i]]++; } sort(t, t+n); for(int i=n-1; i>=0; --i) { k--; zl[t[i]]--; wyn++; if(k==0) { wyn=wyn+zl[t[i]]; cout << wyn; return 0; } } cout << wyn; return 0; } |