牛客假日團隊賽19 D-Chocolate Milk(拓撲排序+樹)

傳送門

思路：

給每個入度爲零的點一些流量，其流量等於該點的出度值。然後拓撲排序，如果某個點的流量等於全部流量，即爲答案點(除起點)。
因爲題目給的是一顆樹，所以不用考慮流量分流後在聚集到一點上。也就是說，除入度爲零的點以外，出度大於1的點及其後面的點都不會是答案。

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> piir;
typedef long long ll;
const int maxn = 1e5+5;
const int INF  = 0x3f3f3f3f;

vector<int> G[maxn];
int in[maxn],flow[maxn],out[maxn];
int n,all;
queue<int> q;

void top(){
    all = 0;
    for(int i=1;i<=n;i++){
        if(in[i]==0){
            in[i]=-1;
            q.push(i);
            flow[i]=G[i].size();
            all += flow[i];
        }
    }
    while(!q.empty()){
        int u=q.front();q.pop();
        for(auto v : G[u]){
            if(out[u]<=1)
                flow[v] += flow[u];
            in[v]--;
            if(in[v]==0) q.push(v);
        }
    }
}
int main(){
    //freopen("in.txt","r",stdin);
    scanf("%d",&n);
    for(int u,v,i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        in[v]++;
        out[u]++;
    }
    top();
    for(int i=1;i<=n;i++){
        if(flow[i]==all && in[i]!=-1)
            printf("%d\n",i);
    }
    return 0;
}