ZOJ-2587 Unique Attack(最小割是否唯一判定)

#include<iostream>
#include<cstdio>
#include<cmath>
#include<map>
#include<set>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstdlib>
#include<cstring>
#include<vector>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef unsigned int uI;
typedef double db;
#define maxn 805
#define inf 0x3f3f3f3f
struct Edge
{
    int to,cap,next;
}edge[50000];

int cnt,head[maxn],d[maxn],s,t;

inline void add(int u,int v,int w)
{
    edge[cnt].to=v;
    edge[cnt].cap=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
    edge[cnt].to=u;
    edge[cnt].cap=w;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}

inline bool bfs()   //BFS构造层次图
{
    int u,i;
    queue<int> q;
    q.push(s);
    memset(d,0,sizeof(d));
    d[s]=1;
    while(!q.empty())
    {
        u=q.front();q.pop();
        if(u==t) return true;   //汇点的层次被算出即可停止，因为根据DFS的规则，和汇点同层或更下一层的结点是不可能走到汇点的。
        for(i=head[u];~i;i=edge[i].next){
            int v=edge[i].to;
            if(!d[v]&&edge[i].cap)
            {
                d[v]=d[u]+1;
                q.push(v);
            }
        }
    }
    return false;
}

int dfs(int u,int a)  //当前结点、目前为止所有弧的最小残量
{
    int flow=0,f,i;
    if(u==t||a==0) return a;
    for(i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(edge[i].cap&&d[v]==d[u]+1)   //满足容量不为0且另一端是下一层次
        {
            f=dfs(v,min(a,edge[i].cap));
            edge[i].cap-=f; //削减路径上各边的容量
            edge[i^1].cap+=f;//添加反向边
            flow+=f;    //增加总流量
            a-=f;   //该结点之前的弧的最小残量削减f后的值
            if(!a) break;   //若为0，则无需再从该结点寻找增广路，因为位于该结点之前的某条弧削减f后会为0，导致无法从源点经过该结点再增广到汇点
        }
    }
    if(flow==0) d[u]=0; //从该点出发找不到增广路，则将该点从层次图中去掉
    return flow;
}

int dinic()
{
    int ans=0;
    while(bfs()) ans+=dfs(s,inf);
    return ans;
}

bool f[805];
int ans;
void DFS1(int u)
{
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(!f[v]&&edge[i].cap)
        {
            f[v]=1;
            ++ans;
            DFS1(v);
        }
    }
}

void DFS2(int u)
{
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(!f[v]&&edge[i^1].cap)
        {
            f[v]=1;
            ++ans;
            DFS2(v);
        }
    }
}
int main()
{
    int a,b,c,n,m;
    while(~scanf("%d%d%d%d",&n,&m,&s,&t)){
        if(!n&&!m&&!s&&!t) break;
        memset(f,0,sizeof(f));
        cnt=0;
        memset(head,-1,sizeof(head));
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        dinic();
        ans=2;
        f[s]=f[t]=1;
        DFS1(s);
        DFS2(t);
        if(ans==n) puts("UNIQUE");
        else puts("AMBIGUOUS");
    }
    return 0;
}