ZOJ 2532 Internship 最小割

题目大意：有一个CIA总部（编号0），n个城市（编号1-n），m个发信号的地方（编号n+1 - n+m）。给你若干条边（u，v，c），意思是信号从u到v（单向），带宽为c。如果你能增加一条边的带宽，使得CIA总部能接收的信号量提高。问，增加哪些边的带宽可以达到这个目的。

首先，你要知道怎么求一个最小割。这个小白书上有写。然后就简单了。白书中只有起点的a[ ]，你再加一个终点开始的类似的数组b[ ]。一条边(u,v)，如果a[u]>0且b[v]>0，则这条边是可行的。

同时，利用这个，也可以求最小割是否唯一。如果存在一个点x，a[x]==0且b[x]==0，则最小割不唯一。

//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime>
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef long long LL;
//typedef unsigned __int64 ULL;
/* ****************** */
const int INF=1000111222;
const double INFF=1e100;
const double eps=1e-8;
const LL mod=1000000007;
const int NN=1005;
const int MM=1010;
/* ****************** */

struct G
{
    int u,v,cap,next;
}E[MM*2];
int p[NN],T;
int dd[NN],tp[NN],qw[NN];
int cs[NN],ct[NN];

void add(int u,int v,int f)
{
    E[T].u=u;
    E[T].v=v;
    E[T].cap=f;
    E[T].next=p[u];
    p[u]=T++;

    E[T].v=u;
    E[T].cap=0;
    E[T].next=p[v];
    p[v]=T++;
}

bool find_path(int st,int en,int n)
{
    int u,v,i,tail,head;
    for(i=0;i<=n;i++)
        dd[i]=-1;
    qw[head=tail=0]=st;
    dd[st]=0;
    while(head<=tail)
    {
        u=qw[head++];
        for(i=p[u];i+1;i=E[i].next)
        {
            v=E[i].v;
            if(dd[v]==-1 && E[i].cap>0)
            {
                dd[v]=dd[u]+1;
                qw[++tail]=v;
            }
        }
    }
    return dd[en]!=-1;
}

int dfs_flow(int u,int &en,int f)
{
    if(u==en || f==0)
        return f;
    int temp,flow=0;
    for( ; tp[u]+1 ; tp[u] = E[tp[u]].next)
    {
        G &e=E[tp[u]];
        if(dd[e.v]==dd[u]+1)
        {
            temp=dfs_flow(e.v,en,min(f,e.cap));
            if(temp>0)
            {
                f-=temp;
                flow+=temp;
                e.cap-=temp;
                E[tp[u]^1].cap+=temp;
                if(f==0)
                    break;
            }
        }
    }
    return flow;
}

int dinic(int st,int en,int n)
{
    int i,ans=0;
    while( find_path(st,en,n) )
    {
        for(i=0;i<=n;i++)
            tp[i]=p[i];
        ans+=dfs_flow(st,en,INF);
    }
    return ans;
}

void dfs1(int u)
{
    int i,v;
    for(i=p[u];i+1;i=E[i].next)
    {
        v=E[i].v;
        if(E[i].cap>0 && cs[v]==0)
        {
            cs[v]=1;
            dfs1(v);
        }
    }
}

void dfs2(int u)
{
    int i,v;
    for(i=p[u];i+1;i=E[i].next)
    {
        v=E[i].v;
        if(E[i^1].cap>0 && ct[v]==0)
        {
            ct[v]=1;
            dfs2(v);
        }
    }
}

int main()
{
    int n,m,l,i;
    int u,v,c;
    bool fg;
    while(scanf("%d%d%d",&n,&m,&l)!=EOF)
    {
        if(n==0)
            break;
        memset(p,-1,sizeof(p));
        T=0;
        for(i=0;i<l;i++)
        {
            scanf("%d%d%d",&u,&v,&c);
            add(u,v,c);
        }
        for(i=1;i<=n;i++)
            add(n+m+1,i,INF);

        c=dinic(n+m+1,0,n+m+1);
     //   cout<<"max_f=="<<c<<endl;

        memset(cs,0,sizeof(cs));
        cs[n+m+1]=1;
        dfs1(n+m+1);

        memset(ct,0,sizeof(ct));
        ct[0]=1;
        dfs2(0);

      //  for(i=0;i<=n+m+1;i++)
       //     printf("%d%c",cs[i],i==n+m+1?'\n':' ');

        fg=false;
        for(i=0;i<T;i+=2)
        {
            u=E[i].u;
            v=E[i].v;
            if(cs[u]==1 && ct[v]==1)
            {
                if(fg)
                    printf(" %d",i/2+1);
                else
                {
                    fg=true;
                    printf("%d",i/2+1);
                }
            }
        }
        puts("");
    }
    return 0;
}