#include <iostream> #include <set> #include <unordered_map> struct dvalue { int value; public: dvalue(int val) : value(val) {} bool operator<(const dvalue& obj2) const { return value > obj2.value; } bool operator>(const dvalue& obj2) const { return value < obj2.value; } bool operator==(const dvalue& obj2) const { return value == obj2.value; } }; std::unordered_map<int, int> count_dval; int main() { std::ios_base::sync_with_stdio(0); std::cin.tie(0); int value = 0; std::set<dvalue> secik; int n, k; std::cin >> n >> k; for (int i = 0; i < n; i++) { int value; std::cin >> value; secik.emplace(value); if (count_dval.count(value)) { count_dval[value] = count_dval[value] + 1; } else count_dval[value] = 1; } for (dvalue val : secik) { if (value < k) { value += count_dval[val.value]; } else { break; } } std::cout << value << std::endl; 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 51 52 53 | #include <iostream> #include <set> #include <unordered_map> struct dvalue { int value; public: dvalue(int val) : value(val) {} bool operator<(const dvalue& obj2) const { return value > obj2.value; } bool operator>(const dvalue& obj2) const { return value < obj2.value; } bool operator==(const dvalue& obj2) const { return value == obj2.value; } }; std::unordered_map<int, int> count_dval; int main() { std::ios_base::sync_with_stdio(0); std::cin.tie(0); int value = 0; std::set<dvalue> secik; int n, k; std::cin >> n >> k; for (int i = 0; i < n; i++) { int value; std::cin >> value; secik.emplace(value); if (count_dval.count(value)) { count_dval[value] = count_dval[value] + 1; } else count_dval[value] = 1; } for (dvalue val : secik) { if (value < k) { value += count_dval[val.value]; } else { break; } } std::cout << value << std::endl; return 0; } |