ZOJ2587 Unique Attack 【最小割的唯一性判定】

题意

给定一个无向图，要求判定分离两个点的最小割是否唯一。

解法

在求出最大流的后，在残余网络中，从源点进行一次搜索，搜索按照未饱和的边进行，得到顶点子集S的顶点个数；再从汇点反向搜索未饱和的边，得到子集T的顶点个数，判定顶点数相加是否等于总共的顶点数。
如果能到所有顶点，则是唯一的，否则不是唯一

参考代码

#include<iostream>
#include<vector>
#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
#define in rad()
inline int rad(){
	int x=0,f=1;char c=getchar();while(c>'9'||c<'0')c=getchar();
	if(c=='-')f=-1,c=getchar();while(c>='0'&&c<='9')x=x*10+c-48,c=getchar();
	return x*f;
}
const int maxn=1e3+10;
const int maxm=4e4+100;
const int inf=1e9;
struct Edge {
	int u,v,w,nxt;
} e[maxm];
int first[maxn]= {},cnt=1,S,T;
inline void add(int u,int v,int w) {
	e[++cnt].v=v;e[cnt].u=u;e[cnt].w=w;e[cnt].nxt=first[u];first[u]=cnt;
}
inline void ins(int u,int v,int w){
	add(u,v,w);add(v,u,0);
}
int N,M,A,B;
namespace D{
	int dis[maxn],cur[maxn];
	int bfs(int s,int t){
		memset(dis,-1,sizeof(dis));
		queue<int>q;
		dis[s]=0;q.push(s);
		while(!q.empty()){
			int u=q.front();q.pop();
			for(int i=first[u];i;i=e[i].nxt){
				int v=e[i].v;
				if(dis[v]!=-1||e[i].w==0)continue;
				dis[v]=dis[u]+1;
				q.push(v);
				if(v==t)return 1;
			}
		}
		return 0;
	} 
	int dfs(int u,int t,int f){
		if(!f||u==t)return f;
		int w,used=0;
		for(int &i=cur[u];i;i=e[i].nxt){
			int v=e[i].v;
			if(dis[v]!=dis[u]+1||e[i].w==0)continue;
			w=dfs(v,t,min(f,e[i].w));
			used+=w;f-=w;
			e[i].w-=w;e[i^1].w+=w;
			if(f==0)break;
		}
		return used;
	}
	int dinic(int s,int t){
		int ret=0;
		while(bfs(s,t)){
			memcpy(cur,first,sizeof(first));
			ret+=dfs(s,t,inf);
		}
		return ret;
	}
}
int vis[maxn];
void dfs(int u,int &cnt,int d){ 
	for(int i=first[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(e[i^d].w && !vis[v]){
			cnt++;vis[v]=1;	dfs(v,cnt,d);
		}
		
	}
}

int solve() {
    int cnta = 0, cntb = 0; // 分别为从两个方向进行搜索而计数
    memset(vis, 0, sizeof (vis));
    vis[S] = vis[T] = 1;
    dfs(S, cnta , 0);//正向边
    memset(vis, 0, sizeof (vis));
    vis[S] = vis[T] = 1;
    dfs(T, cntb, 1);//反向边搜索
    return cnta + cntb == N;
}
int main(){
	S=810,T=811;
	while (scanf("%d %d %d %d", &N, &M, &A, &B), N|M|A|B) {
        int a, b, c;
        cnt = 1;
        memset(first,0, sizeof (first));
        ins(S,A,inf);
        ins(B,T,inf);
        for (int i = 0; i < M; ++i) {
            scanf("%d %d %d", &a, &b, &c);
            add(a, b, c), add(b, a, c);
        }
        D::dinic(S,T);
        printf(solve() ? "UNIQUE\n" : "AMBIGUOUS\n");
    }
	return 0;
}