#include<bits/stdc++.h> using namespace std; int n, k, a, p; int tab[2050]; bool comp(int x, int y) { return (x>y); } int main() { scanf("%d%d", &n, &k); for(int i=0; i<n; i++) { scanf("%d", &tab[i]); } sort(tab, tab+n, comp); a=tab[k-1]; p=k-1; for(int i=n; i>=1; i/=2) { while(p+i<n && tab[p+i]>=a)p+=i; } printf("%d", p+1); }
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 | #include<bits/stdc++.h> using namespace std; int n, k, a, p; int tab[2050]; bool comp(int x, int y) { return (x>y); } int main() { scanf("%d%d", &n, &k); for(int i=0; i<n; i++) { scanf("%d", &tab[i]); } sort(tab, tab+n, comp); a=tab[k-1]; p=k-1; for(int i=n; i>=1; i/=2) { while(p+i<n && tab[p+i]>=a)p+=i; } printf("%d", p+1); } |