题解 P1345 【[USACO5.4]奶牛的电信Telecowmunication】

2017-04-03 12:44:35


这里提供另一种思路

将点x+n和y,y+n和x连边权为INF

再将每一个点i和i+n连边权为1的边

以c1+n为源点,c2为汇点跑最大流就可以了

#include <stdio.h>
#include <string.h>
#include <queue>
#include <string>
#define maxn 400000
#define min(x,y) x<y?x:y
#define INF 0x7f7f7f7f
#define fill(x,i) memset(x,i,sizeof(x))
using namespace std;
struct edge{int y,w,next;}e[maxn];
int ls[maxn],n,m,maxE=1,S,E,state[maxn];
bool vis[maxn];
inline int read()
{
    int x=0,p=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')p=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*p;
}
int add(int x,int y,int w)
{
    e[++maxE]=(edge){y,w,ls[x]};
    ls[x]=maxE;
    e[++maxE]=(edge){x,0,ls[y]};
    ls[y]=maxE;
}
int bfs(int x)
{
    queue <int> t;
    t.push(x);
    fill(state,0);
    state[x]=1;
    while (!t.empty())
    {
        int tt=t.front();t.pop();
        for (int i=ls[tt];i;i=e[i].next)
        {
            if (e[i].w>0&&!state[e[i].y])
            {
                state[e[i].y]=state[tt]+1;
                t.push(e[i].y);
                if (e[i].y==E)
                    return true;
            }
        }
    }
    return false;
}
int cur[maxn];
int find(int x,int mn)
{
    if (x==E||!mn) return mn;
    int ret=0;
    for (int &i=cur[x];i;i=e[i].next)
        if (state[x]+1==state[e[i].y]&&e[i].w>0)
        {
            int d=find(e[i].y,min(mn-ret,e[i].w));
            e[i].w-=d;
            e[i^1].w+=d;
            ret+=d;
            if (ret==mn) break;
        }
    return ret;
}
int main()
{
    n=read(),m=read(),S=read(),E=read();
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read();
        add(x+n,y,INF);
        add(y+n,x,INF);
    }
    for (int i=1;i<=n;i++) add(i,i+n,1);
    int ans=0;
    while (bfs(S+n))
    {
        for (int i=0;i<=E+maxE;i++) 
            cur[i]=ls[i];
        ans+=find(S+n,INF);
    }
    printf("%d\n",ans);
}