ZOJ 2587 Unique Attack

题意：判断最小割是否唯一。
思路（转）：判断最小割是否唯一，先求一次最大流，然后在残留网络中分别从源汇开始dfs一次，找出最小割[S,T]，如果SUT不包含所有点，那么最小割不唯一。假设点i不被SUT包含，那么残留网络中s不能到达i,i不能到达t,即进入i的边和从i出去的边都满流，假设某条进入i的边x满流，这些流量从若干条边y流出i，那么，如果选x为割边，或者选所有对应的y为割边，不会影响最大流，即最小割容量不变，最小割也就不唯一。

#include<stdio.h>
#include<cstring>
#include<math.h>
#define inf 1000000000
#define min(a,b) ((a)>(b))?(b):(a)
using namespace std;
int a,b;
int n,k,dist[1000],gap[1000],edgeHead[1000];//双向边
bool vis[1000];
struct{
    int v,cap,next,re;
}edge[23000];

void addEdge(int u,int v,int ca){
    edge[k].v=v;
    edge[k].cap=ca;
    edge[k].next=edgeHead[u];
    edge[k].re=k + 1;               //这个用来记录此边的反边
    edgeHead[u]=k ++;
    edge[k].v=u;
    edge[k].cap=ca;                 //这里是双向边
    edge[k].next=edgeHead[v];
    edge[k].re=k - 1;
    edgeHead[v]=k ++;
}
int dfs (int p , int limit)
{
    if(p==b)return limit;

    for(int i=edgeHead[p];i!=0;i=edge[i].next){
        int v = edge[i].v;
        if(dist[p]==(dist[v]+1) && edge[i].cap>0){
            int t=dfs(v,min(limit , edge[i].cap));
            if(t<0)return t;//没有增广路了
            if(t>0)//更新流
            {
                edge[i].cap-=t;
                edge[edge[i].re].cap+=t;
                return t;
            }
        }
    }

    int tmp=n+1;
    for(int i=edgeHead[p];i!=0;i=edge[i].next){
        int v = edge[i].v;
        if(edge[i].cap>0)
            tmp=min(tmp,dist[v]+1);
    }

    if(--gap[dist[p]]==0 || dist[0]>n)return -1;//出现断层或回流已满
    ++gap[dist[p]=tmp];
    return 0;
}

int SAP()
{
    gap[a]=n+1;
    int f = 0 , t=0;
    while (~(t=dfs(a,inf))) f+=t;
    return f;
}
void dfs2(int s){
	int i,v;
	vis[s]=1;
	for(i=edgeHead[s];i!=0;i=edge[i].next){
		if(edge[edge[i].re].cap>0 && !vis[edge[i].v])   //注意这里搜的是能够到达汇点的点
			dfs2(edge[i].v);
	}
}
void dfs1(int s){
	int i,v;
	vis[s]=1;
	for(i=edgeHead[s];i!=0;i=edge[i].next){
		if(edge[i].cap>0 && !vis[edge[i].v])
			dfs1(edge[i].v);
	}
}
int main(){
	int i,j,e,u,v,w;
	while(scanf("%d %d %d %d",&n,&e,&a,&b) && !(n==0 && e==0 && a==0 && b==0)){
		memset(edgeHead,0,sizeof(edgeHead));
		memset(dist,0,sizeof(dist));
		memset(vis,0,sizeof(vis));
		memset (gap , 0 , sizeof(gap));
		k=1;

		for(i=1;i<=e;i++){
			scanf("%d %d %d",&u,&v,&w);
			addEdge(u,v,w);
		}
		int k=SAP();
		dfs1(a);
		dfs2(b);
		for(i=1;i<=n;i++){
			if(!vis[i]){
				break;
			}
		}
		if(i<=n)
			printf("AMBIGUOUS\n");
		else
			printf("UNIQUE\n");
	}
}