#include<bits/stdc++.h> using namespace std; int tab[1000005]; int cmp(int a, int b){ return a>b; } int main(){ int n, k; cin>>n>>k; for(int i=0;i<n;i++){ cin>>tab[i]; } sort(tab, tab+n, cmp); //for(int i=0;i<n;i++){ // cout<<tab[i]<<" "; //} int wynik=0; for(int i=k;i<n;i++){ if(tab[i]==tab[k-1]){ wynik++; } } cout<<wynik+k; 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 | #include<bits/stdc++.h> using namespace std; int tab[1000005]; int cmp(int a, int b){ return a>b; } int main(){ int n, k; cin>>n>>k; for(int i=0;i<n;i++){ cin>>tab[i]; } sort(tab, tab+n, cmp); //for(int i=0;i<n;i++){ // cout<<tab[i]<<" "; //} int wynik=0; for(int i=k;i<n;i++){ if(tab[i]==tab[k-1]){ wynik++; } } cout<<wynik+k; return 0; } |