首页 > 代码库 > hdu3452 无向树去掉最小的边集使任何叶子与根不连通 / 最小割

hdu3452 无向树去掉最小的边集使任何叶子与根不连通 / 最小割

思路一下就上来了,叶子向汇点连边,inf保证不会成为割,跑根到汇点最小割即可。注意无向树双向建边。基础题,分分钟1A:

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<set>
#include<vector>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxv=1005,maxe=10000;
int nume=0;int head[maxv];int e[maxe][3];
void inline adde(int i,int j,int c)
{
    e[nume][0]=j;e[nume][1]=head[i];head[i]=nume;
    e[nume++][2]=c;
    e[nume][0]=i;e[nume][1]=head[j];head[j]=nume;
    e[nume++][2]=0;
}
int ss,tt,n,m;
int vis[maxv];int lev[maxv];
bool bfs()
{
    for(int i=0;i<maxv;i++)
      vis[i]=lev[i]=0;
    queue<int>q;
    q.push(ss);
    vis[ss]=1;
    while(!q.empty())
    {
        int cur=q.front();
        q.pop();
        for(int i=head[cur];i!=-1;i=e[i][1])
        {
            int v=e[i][0];
            if(!vis[v]&&e[i][2]>0)
            {
                lev[v]=lev[cur]+1;
                vis[v]=1;
                q.push(v);
            }
        }
    }
    return vis[tt];
}
int dfs(int u,int minf)
{
    if(u==tt||minf==0)return minf;
    int sumf=0,f;
    for(int i=head[u];i!=-1&&minf;i=e[i][1])
    {
        int v=e[i][0];
        if(lev[v]==lev[u]+1&&e[i][2]>0)
        {
            f=dfs(v,minf<e[i][2]?minf:e[i][2]);
            e[i][2]-=f;e[i^1][2]+=f;
            sumf+=f;minf-=f;
        }
    }
    if(!sumf) lev[u]=-1;
    return sumf;
}
int dinic()
{
    int sum=0;
    while(bfs())sum+=dfs(ss,inf);
    return sum;
};
int ind[maxv];
void read_build()
{
    int aa,bb,cc;
    for(int i=0;i<n-1;i++)
     {
        scanf("%d%d%d",&aa,&bb,&cc);
        adde(aa,bb,cc);
        adde(bb,aa,cc);
        ind[aa]++;ind[bb]++;
     }
     for(int i=1;i<=n;i++)
        if(i!=m&&ind[i]==1)
        {
            adde(i,tt,inf);
        }

  /*  for(int i=0;i<=tt;i++)
      for(int j=head[i];j!=-1;j=e[j][1])
      {
          if(j%2==0)
          printf("%d->%d:%d\n",i,e[j][0],e[j][2]);
      }*/
}
void init()
{
    nume=0;
    ss=m;tt=n+1;
    for(int i=0;i<=tt;i++)
      {
          head[i]=-1;ind[i]=0;
      }
}
int main()
{
      while(~scanf("%d%d",&n,&m)&&(n||m))
     {
         init();
        read_build();
       int ans;
       ans=dinic();
      printf("%d\n",ans);
    }
    return 0;
}