#include <iostream>
#include <map>
#include <functional>
using namespace std;
int get(map<int, int, greater<int>>* fishes, int num) {
try {
// Found the item, return it
return fishes->at(num);
}
// Didn't find the item
catch(out_of_range) {
// Return 0
return 0;
}
}
void change(map<int, int, greater<int>>* fishes, int num, int by) {
// Change the number
fishes->insert_or_assign(num, get(fishes, num) + by);
}
int main() {
ios_base::sync_with_stdio();
cin.tie();
cout.tie();
// Get fish count
int fishCount;
cin >> fishCount;
// Map of fish weights to fish count
map<int, int, greater<int>> fishes;
// Load fishes
for (int i = 0; i < fishCount; i++) {
int weight;
cin >> weight;
// Increment count
change(&fishes, weight, 1);
}
// Get event count
int eventCount;
cin >> eventCount;
// Load events
for (int i = 0; i < eventCount; i++) {
// Read event data
int type, weight, target;
cin >> type >> weight;
// Type 1 events
if (type == 1) {
// Get the target
cin >> target;
// Find the best matches
for (pair<int, int> pair : fishes) {
// Check the weight
}
}
// Type 2 and 3 events
else {
// Apply the changes
change(&fishes, weight, type == 2 ? 1 : -1);
}
}
}
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | #include <iostream> #include <map> #include <functional> using namespace std; int get(map<int, int, greater<int>>* fishes, int num) { try { // Found the item, return it return fishes->at(num); } // Didn't find the item catch(out_of_range) { // Return 0 return 0; } } void change(map<int, int, greater<int>>* fishes, int num, int by) { // Change the number fishes->insert_or_assign(num, get(fishes, num) + by); } int main() { ios_base::sync_with_stdio(); cin.tie(); cout.tie(); // Get fish count int fishCount; cin >> fishCount; // Map of fish weights to fish count map<int, int, greater<int>> fishes; // Load fishes for (int i = 0; i < fishCount; i++) { int weight; cin >> weight; // Increment count change(&fishes, weight, 1); } // Get event count int eventCount; cin >> eventCount; // Load events for (int i = 0; i < eventCount; i++) { // Read event data int type, weight, target; cin >> type >> weight; // Type 1 events if (type == 1) { // Get the target cin >> target; // Find the best matches for (pair<int, int> pair : fishes) { // Check the weight } } // Type 2 and 3 events else { // Apply the changes change(&fishes, weight, type == 2 ? 1 : -1); } } } |
English