ECOO '20 P2 - Online Shopping

    1#include <bits/stdc++.h>
    2 
    3using namespace std;
    4 
    5int main(){
    6    ios::sync_with_stdio(0);
    7    cin.tie(0);
    8    int t;
    9    cin >> t;
    10    while(t--){
    11        int n;
    12        cin >> n;
    13        unordered_map<string, vector<pair<int, int>>> pos;
    14        for(int i = 0; i < n; i++){
    15            int m;
    16            cin >> m;
    17            for(int j = 0; j < m; j++){
    18                string s; int c, v;
    19                cin >> s >> c >> v;
    20                pos[s].emplace_back(c, v);
    21            }
    22        }
    23        int k;
    24        cin >> k;
    25        unordered_map<string, int> need;
    26        for(int i = 0; i < k; i++){
    27            string s; int v;
    28            cin >> s >> v;
    29            need[s] = v;
    30        }
    31        int cost = 0;
    32        for(auto &[s, v] : pos){
    33            sort(v.begin(), v.end());
    34        }
    35        for(auto[name, amount] : need){
    36            auto &vec = pos[name];
    37            for(auto[co, am] : vec){
    38                if(amount >= am){
    39                    cost += co * am;
    40                    amount -= am;
    41                } else {
    42                    cost += co * amount;
    43                    break;
    44                }
    45            }
    46        }
    47        cout << cost << '\n';
    48    }
    49}