zoj2587 sap最小割+判断割的唯一性

http://www.cnblogs.com/ka200812/archive/2011/07/30/2121872.html

判断割的唯一性可参考上文章

寻找割

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <math.h>
#include <vector>
#include <cstdio>
#include <string>
#include<string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 999999999;
#define set(x,i) memset(x,i,sizeof(x));
const int maxn=200000*5; //边的数量，应该比正常边的数量多两倍
const int MAXN=20010;//节点数

int s,t,tot;
int cur[MAXN];//当前弧优化
int pre[MAXN];//先祖
int level[MAXN];//高度
int gap[MAXN];//gap优化
bool vis[MAXN];
struct EDGE 
{
	int to,cost,next;
}edge[maxn];
int head[maxn],cnt;

template<class T>
inline void read(T &m)
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	m=x*f;
}
inline void addedge(int a,int b,int c)
{
	edge[cnt].to=b;
	edge[cnt].cost=c;
	edge[cnt].next=head[a];
	head[a]=cnt++;

	edge[cnt].to=a;
	edge[cnt].cost=0;
	edge[cnt].next=head[b];
	head[b]=cnt++;
}
void build_graph(int n,int m) //节点数(1->n)，边数  
{
	tot=n+2,cnt=0;//cnt=0为了更好的异或
	set(pre,-1);
	set(gap,0);
	set(level,0);
	set(head,-1);
	gap[0]=tot;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		scanf("%d %d %d",&a,&b,&c);
		addedge(a,b,c);
		addedge(b,a,c);
	}
}
int sap()  //无需改动
{
	int u=s,ans=0,inf=INF;
	for(int i=0; i<tot; i++)//初始化当前弧为第一条弧 
		cur[i]=head[i];
	while(level[s]<tot)
	{
		bool flag=0;
		for(int &i=cur[u];i!=-1;i=edge[i].next) //读取边，并更新当前弧
		{
			int v=edge[i].to;
			if(edge[i].cost>0&&level[u]==level[v]+1)
			{
				flag=1;
				if(edge[i].cost<inf)
					inf=edge[i].cost;
				pre[v]=u;
				u=v;
				if(u==t)
				{
					while(u!=s) //结束后u==s
					{
						u=pre[u];
						edge[cur[u]].cost-=inf;
						edge[cur[u]^1].cost+=inf;//异或是找与其配对的边
					}
					ans+=inf;
					inf=INF;
				}
				break;
			}
		}
		if(!flag)
		{
			if(--gap[level[u]]==0)//gap优化
				break;
			int minn=tot;
			for(int i=head[u];i!=-1;i=edge[i].next)
			{
				int v=edge[i].to;
				if(edge[i].cost>0&&level[v]<minn)
				{
					minn=level[v];
					cur[u]=i;//更新当前弧，让其从度最小的线段开始
				}
			}
			level[u]=minn+1;
			gap[minn+1]++;
			if(u!=s)
				u=pre[u];
		}
	}
	return ans;
}
void dfs(int &sum,int k,int l)
{
	vis[k]=1;
	sum++;
	for(int i=head[k];i!=-1;i=edge[i].next)
		if(!vis[edge[i].to]&&edge[i^l].cost)  //同时也可为寻找子集,只有弧没满,才属于子集并继续搜索。
			dfs(sum,edge[i].to,l);
}
int main()
{
	int n,m;
	while(read(n),read(m),read(s),read(t),n+m+s+t!=0)
	{
		build_graph(n,m);
		sap();
		set(vis,0);
		int sum=0;
		dfs(sum,s,0);  //搜索
		dfs(sum,t,1);
		if(sum==n)
			cout<<"UNIQUE\n";
		else
			cout<<"AMBIGUOUS\n";
	}
	return 0;
}