#include<bits/stdc++.h> using namespace std; bool cmp(int a,int b){ return a>b; } int main(){ ios::sync_with_stdio(0); cin.tie(0); int n,k; cin>>n>>k; k--; vector<int> t(n); for(int i=0;i<n;i++){ cin>>t[i]; } sort(t.begin(),t.end(),cmp); int ans=k; ans++; for(int i=k+1;i<n;i++){ if(t[i]==t[k]){ ans++; } } cout<<ans<<"\n"; }
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 | #include<bits/stdc++.h> using namespace std; bool cmp(int a,int b){ return a>b; } int main(){ ios::sync_with_stdio(0); cin.tie(0); int n,k; cin>>n>>k; k--; vector<int> t(n); for(int i=0;i<n;i++){ cin>>t[i]; } sort(t.begin(),t.end(),cmp); int ans=k; ans++; for(int i=k+1;i<n;i++){ if(t[i]==t[k]){ ans++; } } cout<<ans<<"\n"; } |