#include <bits/stdc++.h> #pragma GCC optimize(-O3) #pragma GCC optimize(Ofast) #pragma GCC optimize("unroll-loops") #define fi first #define se second #define p_b push_back #define pll pair<ll,ll> #define pii pair<int,int> #define m_p make_pair #define all(x) x.begin(),x.end() #define sset ordered_set #define sqr(x) (x)*(x) #define pw(x) (1ll << (x)) #define sz(x) (int)x.size() #define fout(x) {cout << x << "\n"; return; } using namespace std; typedef long long ll; typedef long double ld; const int MAXN = 1e5; const int M = pw(16); const long long mod = 998244353; const int N = 2e5; const int inf = 1e9; template <typename T> void vout(T s){cout << s << endl;exit(0);} int main(){ ios_base::sync_with_stdio(0); cin.tie(0); int n, k; cin >> n >> k; vector <int> a(n); for(auto &i : a) cin >> i; sort(a.rbegin(), a.rend()); while(k < n && a[k - 1] == a[k]) k++; cout << k << "\n"; 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 38 39 40 41 42 43 44 45 46 47 48 49 50 | #include <bits/stdc++.h> #pragma GCC optimize(-O3) #pragma GCC optimize(Ofast) #pragma GCC optimize("unroll-loops") #define fi first #define se second #define p_b push_back #define pll pair<ll,ll> #define pii pair<int,int> #define m_p make_pair #define all(x) x.begin(),x.end() #define sset ordered_set #define sqr(x) (x)*(x) #define pw(x) (1ll << (x)) #define sz(x) (int)x.size() #define fout(x) {cout << x << "\n"; return; } using namespace std; typedef long long ll; typedef long double ld; const int MAXN = 1e5; const int M = pw(16); const long long mod = 998244353; const int N = 2e5; const int inf = 1e9; template <typename T> void vout(T s){cout << s << endl;exit(0);} int main(){ ios_base::sync_with_stdio(0); cin.tie(0); int n, k; cin >> n >> k; vector <int> a(n); for(auto &i : a) cin >> i; sort(a.rbegin(), a.rend()); while(k < n && a[k - 1] == a[k]) k++; cout << k << "\n"; return 0; } |