第十三届山东省ICPC大学生程序设计竞赛 B. Building Company

题目

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
#define lson p << 1
#define rson p << 1 | 1
const int maxn = 1e6 + 5, inf = 1e18, maxm = 4e4 + 5;
const int N = 1e6;
 const int mod = 1e9 + 7;
// const int mod = 998244353;
//const __int128 mod = 212370440130137957LL;
// int a[505][5005];
// bool vis[505][505];
int n, m;
int a[maxn];

string s;

struct Node{
    // int val, id;
    // bool operator<(const Node &u)const{
    //     return val < u.val;//!!!!!!!!!!!!!!!!!!
    // }
    int v, w;
};

//long long ? maxn ? n? m?
void solve(){
    int res = 0;
    int g;
    cin >> g;
    map<int, int> cnt;
    for(int i = 1; i <= g; i++){
        int x, y;
        cin >> x >> y;
        cnt[x] = y;
    }
    cin >> n;
    int m, k;
    int x, y;
    vector<int> deg(n + 5, 0);//deg[i]为第i个项目的度数，当度数为0表示该项目可以完成
    map<int, vector<pair<int, int>>> req;//req[x]存 职业类型为x的项目要求，pair.first为需要类型x的人数，pair.second为 该要求 所属的项目编号
    queue<int> q;
    vector<vector<pair<int, int>>> add(n + 5);//add[i]存 项目i完成后可以增加的 <类型，人数>
    for(int i = 1; i <= n; i++){
        cin >> m;
        for(int j = 1; j <= m; j++){
            cin >> x >> y;
            if(cnt[x] < y){
                deg[i]++;
                req[x].pb({y, i});
            }
        }
        if(!deg[i]){
            q.push(i);
        }
        cin >> k;
        for(int j = 1; j <= k; j++){
            cin >> x >> y;
            add[i].pb({x, y});
        }
    }
    for(auto &[x, y] : req){//必须加&，不然就是无效排序
        sort(y.begin(), y.end(), greater<>());
    }
    while(!q.empty()){
        int u = q.front();
        q.pop();
        res++;
        for(auto [x, y] : add[u]){
            cnt[x] += y;
            while(!req[x].empty() && req[x].back().first <= cnt[x]){//类似双指针，因为当前只改变了cnt[x]，
                                                                    //只用在类型x的要求中看能不能完成
                int v = req[x].back().second;
                deg[v]--;
                if(!deg[v]){
                    q.push(v);
                }
                req[x].pop_back();
            }
        }
    }
    cout << res << '\n';
}   
    
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int T = 1;
    // cin >> T;
    while (T--)
    {
        solve();
    }
    return 0;
}