#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; } |