gym102394 E. Exchanging Gifts（拓扑排序+map）

题意：

解法：

难点在于如何计算题目的g[].
如果能计算出g[],设g[]中的最大值为ma,总和为sum,
如果ma*2<=sum,那么所有人都可以换到,答案为sum,
否则有ma*2-sum的换不到,答案为sum-(ma*2-sum).

g[]是由其他数组合并得来,
发现对于(2,x,y)操作,建立的是有向无环图,
因此考虑拓扑排序,计算出每个终止结点被统计了多少次(递推出cnt[]),
最后一起加到g[]中就行了.

code：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=2e6+5;
vector<int>f[maxm];
vector<int>g[maxm];
unordered_map<int,ll>res;
bool mark[maxm];
bool vis[maxm];
ll cnt[maxm];
int in[maxm];
int n;
void topo(){//拓扑排序递推出cnt[]
    queue<int>q;
    cnt[n]=1;
    for(int i=1;i<=n;i++){
        if(!in[i]){
            q.push(i);
        }
    }
    while(q.size()){
        int x=q.front();q.pop();
        for(int v:g[x]){
            if(in[v]>0){
                in[v]--;
                cnt[v]+=cnt[x];
                if(!in[v]){
                    q.push(v);
                }
            }
        }
    }
}
void solve(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)f[i].clear();
    for(int i=1;i<=n;i++)g[i].clear();
    for(int i=1;i<=n;i++)in[i]=0;
    for(int i=1;i<=n;i++)cnt[i]=0;
    for(int i=1;i<=n;i++)mark[i]=0;
    for(int i=1;i<=n;i++){
        int op;scanf("%d",&op);
        if(op==1){
            int k;scanf("%d",&k);
            for(int j=1;j<=k;j++){
                int x;scanf("%d",&x);
                f[i].push_back(x);
            }
            mark[i]=1;
        }else{
            int x,y;scanf("%d%d",&x,&y);
            g[i].push_back(x);
            g[i].push_back(y);
            in[x]++,in[y]++;
        }
    }
    topo();
    res.clear();
    for(int i=1;i<=n;i++){
        if(mark[i]&&cnt[i]){
            for(auto j:f[i]){
                res[j]+=cnt[i];
            }
        }
    }
    ll sum=0;
    ll ma=0;
    for(auto &i:res){
        sum+=i.second;
        ma=max(ma,i.second);
    }
    if(ma*2<=sum){
        printf("%lld\n",sum);
    }else{
        printf("%lld\n",sum-(ma*2-sum));
    }
}
signed main(){
    int T;scanf("%d",&T);
    while(T--){
        solve();
    }
    return 0;
}

---