bzoj3504: [Cqoi2014]危桥--最大流

题目大意：给张无向图，有两个人a,b分别从各自的起点走向各自的终点，走A,B个来回，图里有些边只能走两次，求问是否能满足a，b的需求

按照题目给的表建图

S连a1,b1

a2,b2连T

跑最大流看是否满流

为了防止a1跑到b2的情况

第二遍

S连a1,b2

a2,b1连T

若还是满流说明没问题

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#define INF 0x7fffffff
using namespace std;
const int maxe = 5050, maxn = 55;
int n,a1,a2,A,b1,b2,B;
char st[1000];
struct node{
    int to[maxe],next[maxe],flow[maxe];
    int head[maxn],h[maxn],tot,s,t;
    void init(){s=0; t=n+1; memset(head,-1,sizeof(head)); tot=-1;}
    void insert(int u, int v, int f){
        to[++tot]=v;
        next[tot]=head[u];
        flow[tot]=f;
        head[u]=tot;
        to[++tot]=u;
        next[tot]=head[v];
        flow[tot]=0;
        head[v]=tot;
    }
    bool bfs(){
        memset(h,-1,sizeof(h));
        queue<int> Q;
        Q.push(s);
        h[s]=0;
        while (!Q.empty()){
            int now=Q.front();
            Q.pop();
            for (int i=head[now]; i!=-1; i=next[i]){
                int v=to[i];
                if (flow[i] && h[v]==-1){
                    h[v]=h[now]+1;
                    Q.push(v);
                }
            }
        }
        if (h[t]==-1) return 0; return 1;
    }
    int dfs(int x, int mx){
        if (x==t) return mx;
        int res=0;
        for (int i=head[x]; i!=-1; i=next[i]){
            int v=to[i];
            if (flow[i] && h[v]==h[x]+1){
                int f=dfs(v,min(flow[i],mx));
                flow[i]-=f; flow[i^1]+=f;
                res+=f; mx-=f;
                if (!mx) break;
            }
        }
        if (!res) h[x]=-1;
        return res;
    }    
    int maxflow(){
        int ans=0;
        while (bfs()) ans+=dfs(s,INF);
        return ans;
    }
}x,y;
int main(){
    while (scanf("%d%d%d%d%d%d%d", &n, &a1, &a2, &A, &b1, &b2, &B)!=EOF){
        a1++; a2++; b1++; b2++;
        A*=2; B*=2; x.init();
        for (int i=1; i<=n; i++){
            scanf("%s", st+1);
            for (int j=1; j<=n; j++){
                if (st[j]=='O') x.insert(i,j,2);
                else if (st[j]=='N') x.insert(i,j,INF);
            }
        }
        y=x;
        int s=0, t=n+1;
        x.insert(s,a1,A); x.insert(a2,t,A); x.insert(s,b1,B); x.insert(b2,t,B);
        y.insert(s,a1,A); y.insert(a2,t,A); y.insert(s,b2,B); y.insert(b1,t,B);
        if (x.maxflow()==A+B && y.maxflow()==A+B) printf("Yes
"); else printf("No
");
    }
    return 0;
}