P3931 SAC E#1 - 一道难题 Tree

题目描述

冴月麟为了守护幻想乡，而制造了幻想乡的倒影，将真实的幻想乡封印了。任何人都无法进入真实的幻想乡了，但是她给前来救她的魏潇承留了一个线索。

她设置了一棵树（有根）。树的每一条边上具有割掉该边的代价。

魏潇承需要计算出割开这棵树的最小代价，这就是冴月麟和魏潇承约定的小秘密。

帮帮魏潇承吧。

注：所谓割开一棵有根树，就是删除若干条边，使得任何任何叶子节点和根节点不连通。

输入格式

输入第一行两个整数n，S表示树的节点个数和根。

接下来n-1行每行三个整数a、b、c，表示a、b之间有一条代价为c的边。

输出格式

输出包含一行，一个整数，表示所求最小代价。

---

题解：
最小割/最大流，设置一个超级汇点t，让所有叶子结点连向t，跑一遍最大流即可

---

AC代码：

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
#define LL long long
const int MAXN = 4e5+50;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int n,s,t,tot=1,head[MAXN],w[MAXN],nxt[MAXN],to[MAXN],h[MAXN];
vector<pair<int,int> > g[MAXN];
inline void ade(int u,int v,int ww){
    to[++tot]=v; w[tot]=ww; nxt[tot]=head[u]; head[u]=tot;
}
inline void add(int u,int v,int w){ ade(u,v,w); ade(v,u,0); }
inline void DFS(int u,int fa){
    bool flag=false;
    for(int i=0;i<(int)g[u].size();i++){
        int v=g[u][i].first,w=g[u][i].second;
        if(v==fa) continue;
        flag = true; add(u,v,w);
        DFS(v,u);
    }
    if(!flag) add(u,t,INF);
}
inline int bfs(){
    queue<int> que; que.push(s); memset(h,0,sizeof(h)); h[s]=1;
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=head[u];i;i=nxt[i]){
            if(w[i] && !h[to[i]]){
                h[to[i]]=h[u]+1; que.push(to[i]);
            }
        }
    }
    return h[t];
}
inline int dfs(int x,int f){
    if(x==t) return f; int fl=0;
    for(int i=head[x];i;i=nxt[i]){
        if(w[i] && h[to[i]]==h[x]+1){
            int mi=dfs(to[i],min(f,w[i]));
            w[i]-=mi; w[i^1]+=mi; fl+=mi; f-=mi;
        }
    }
    if(!fl) h[x]=-1;
    return fl;
}
inline int dinic(){
    int res=0;
    while(bfs()) res+=dfs(s,INF);
    return res;
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    scanf("%d%d",&n,&s); t=n+1;
    for(int i=1;i<=n-1;i++){
        int u,v,w; scanf("%d%d%d",&u,&v,&w);
        g[u].push_back(make_pair(v,w));
        g[v].push_back(make_pair(u,w));
    }
    DFS(s,0);
    printf("%d\n",dinic());
    return 0;
}