#include<bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k; cin>>n>>k; vector<int> a(n); for(auto& i: a) cin>>i; sort(a.rbegin(), a.rend()); int ans = k; for(int i=k; i<n && a[i]==a[k-1]; ++i) ans++; cout<<ans<<'\n'; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | #include<bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k; cin>>n>>k; vector<int> a(n); for(auto& i: a) cin>>i; sort(a.rbegin(), a.rend()); int ans = k; for(int i=k; i<n && a[i]==a[k-1]; ++i) ans++; cout<<ans<<'\n'; } |