#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
vector<int> vec;
int n{0};
int k{0};
int a{0};
int b{0};
int sum{0};
int main() {
cin>>n;
cin>>k;
for(int i=1;i<=n;i++){
cin>>a;
vec.push_back(a);
}
sort(vec.begin(),vec.end(),greater<int>());
if(k==n){
cout<<k;
}
if(k!=n){
b=vec[k-1];
for(int i=k;i<vec.size();i++){
if(b==vec[i]){
sum+=1;
}
else{
break;
}
}
cout<<k+sum;
}
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 | #include <algorithm> #include <iostream> #include <vector> using namespace std; vector<int> vec; int n{0}; int k{0}; int a{0}; int b{0}; int sum{0}; int main() { cin>>n; cin>>k; for(int i=1;i<=n;i++){ cin>>a; vec.push_back(a); } sort(vec.begin(),vec.end(),greater<int>()); if(k==n){ cout<<k; } if(k!=n){ b=vec[k-1]; for(int i=k;i<vec.size();i++){ if(b==vec[i]){ sum+=1; } else{ break; } } cout<<k+sum; } return 0; } |
English