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;
}