#include <cstdlib> #include <cstring> #include <string> #include <vector> #include <algorithm> #include <map> #include <cstdio> #include <iostream> using namespace std; int n, k, x; int main() { vector<int> v; scanf("%d %d", &n, &k); for(int i=0; i<n; i++){ scanf("%d", &x); v.push_back(x); } sort(v.begin(), v.end(), greater<>()); int min=v[k-1]; for(int i=k; i<n; i++){ if(v[i]==min){ k++; } } printf("%d\n", k); 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 | #include <cstdlib> #include <cstring> #include <string> #include <vector> #include <algorithm> #include <map> #include <cstdio> #include <iostream> using namespace std; int n, k, x; int main() { vector<int> v; scanf("%d %d", &n, &k); for(int i=0; i<n; i++){ scanf("%d", &x); v.push_back(x); } sort(v.begin(), v.end(), greater<>()); int min=v[k-1]; for(int i=k; i<n; i++){ if(v[i]==min){ k++; } } printf("%d\n", k); return 0; } |