#include <iostream> #include <algorithm> using namespace std; int main() { short n, k, answer; cin >> n >> k; answer = k; char arr[n]; for(char& j : arr) scanf("%hhi", &j); sort(arr, arr+n, greater<short>()); char last_given = arr[k-1]; for(char* i = arr+k; i < arr+n && *i == last_given; ++i) answer++; cout << answer << '\n'; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | #include <iostream> #include <algorithm> using namespace std; int main() { short n, k, answer; cin >> n >> k; answer = k; char arr[n]; for(char& j : arr) scanf("%hhi", &j); sort(arr, arr+n, greater<short>()); char last_given = arr[k-1]; for(char* i = arr+k; i < arr+n && *i == last_given; ++i) answer++; cout << answer << '\n'; } |