HDU4289 Control

　　原题传送：http://acm.hdu.edu.cn/showproblem.php?pid=4289

　　网络流，最大流 + 拆点。

　　最大流：sap算法

　　常用的两种拆点方式：

　　　　1. 对于每个节点id，拆成 id<<1 和 id<<1|1 两个点

　　　　2.对于每个节点id，拆成 id 和 id+n 两个点

#include <stdio.h>
#include <string.h>
#define L(u) ((u) << 1)
#define R(u) ((u) << 1 | 1)
#define N 420
#define M 100000
#define INF 0x3f3f3f3f

int gap[N],dis[N],pre[N],cur[N];
int n,m,NE, s, t;
int head[N];
struct Node{
    int c,pos,next;
}E[M];

inline void checkmin(int &a,int b)  {if(a == -1 || a > b)a = b;}

void add_edge(int u,int v,int c)
{
    E[NE].c = c;
    E[NE].pos = v;
    E[NE].next = head[u];   
    head[u] = NE++;
    
    E[NE].c = 0;    // !反向初始为0
    E[NE].pos = u;
    E[NE].next = head[v];   
    head[v] = NE++;
}

int sap()
{
    memset(dis,0,sizeof dis);
    memset(gap,0,sizeof gap);
    memcpy (cur, head, sizeof dis);
    int u=pre[s]=s,maxflow=0,aug=-1;
    gap[0]=n;
    while(dis[s]<n)
    {
loop:for(int &i=cur[u];i!=-1;i=E[i].next)
        {
            int v=E[i].pos;
            if(E[i].c && dis[u]==dis[v]+1)
            {
                checkmin(aug, E[i].c);
                pre[v]=u;
                u=v;
                if(v==t)
                {
                    maxflow+=aug;
                    for(u=pre[u];v!=s;v=u,u=pre[u])
                    {
                        E[cur[u]].c-=aug;
                        E[cur[u]^1].c+=aug;
                    }
                    aug=-1;
                }
                goto loop;
            }
        }
        int mindis=n;
        for(int i=head[u];i!=-1;i=E[i].next)
        {
            int v=E[i].pos;
            if(E[i].c && mindis>dis[v])
            {
                cur[u]=i;
                mindis=dis[v];
            }
        }
        if((--gap[dis[u]])==0) break;
        gap[dis[u]=mindis+1]++;
        u=pre[u];
    }
    return maxflow;
}
 
void init()
{
    memset(head, -1, sizeof head);
    NE = 0;
}
 
int main()
{
    int i, a, b, w;
    while(scanf("%d%d", &n, &m) != EOF)
    {
        init();
        scanf("%d%d", &s, &t);
        add_edge(0, L(s), INF);          // 0为超级源点
        add_edge(R(t), R(n + 1), INF);   // R(n + 1)为超级汇点
        s = 0;
        t = R(n + 1);
        for(i = 1; i <= n; i ++)
        {
            scanf("%d", &w);
            add_edge(L(i), R(i), w);
            add_edge(R(i), L(i), w); 
        }
        for(i = 0; i < m; i ++)
        {
            scanf("%d%d", &a, &b);
            add_edge(R(a), L(b), INF);
            add_edge(R(b), L(a), INF);
        }
        n = 2 * (n + 1);                 // 总节点数的变化
        printf("%d\n", sap());
    }
    return 0;
}