HDU3549(最大流算法的Dinic算法)

题目：Flow Problem

#include <iostream>
#include <stdio.h>
#include <string.h>

using namespace std;

const int N=2050;     //N表示点的数量
const int M=2500000;  //M表示边的数量，注意应大于(N-1)*N，因为网络是双向的
const int INF=0x7fffffff;

int e;
int ver[M],next[M],flow[M];  //ver为边指向的节点，flow为边的容量，next为链表的下一条边
int head[N],work[N],dis[N],q[N]; //head为节点的链表头，work 用于算法中的临时链表头，dis 计算距离

void init()
{
    e=0;
    memset(head,-1,sizeof(head));
}

void add(int u,int v,int c)  //增加一条从u到v容量为c的边
{
    ver[e]=v;flow[e]=c;next[e]=head[u];head[u]=e++;
    ver[e]=u;flow[e]=0;next[e]=head[v];head[v]=e++;
}


//广搜计算出每个点与源点的最短距离，如果不能到达汇点说明算法结束
int bfs(int S,int T)
{
    int rear=0;
    memset(dis,-1,sizeof(dis));
    dis[S]=0;q[rear++]=S;
    for(int i=0;i<rear;i++)
    {
        for(int j=head[q[i]];j!=-1;j=next[j])
        {
            if(flow[j]&&dis[ver[j]]==-1)
            {
                dis[ver[j]]=dis[q[i]]+1;
                q[rear++]=ver[j];
                if(ver[j]==T) return 1;
            }
        }
    }
    return 0;
}

//寻找可行流的增广路算法，按节点的距离来找，加快速度
int dfs(int cur,int a,int T)
{
    if(cur==T) return a;
    for(int &i=work[cur];i!=-1;i=next[i]) //work是临时链表头，这里用i引用它，这样寻找过的边不再寻找
    {
        if(flow[i]&&dis[ver[i]]==dis[cur]+1)
        {
            int t=dfs(ver[i],min(a,flow[i]),T);
            if(t)
            {
                flow[i]-=t;
                flow[i^1]+=t;  //正反向边容量改变
                return t;
            }
        }
    }
    return 0;
}

//求最大流，直到没有可行流
long long Dinic(int S,int T)
{
    long long ans=0;
    while(bfs(S,T))
    {
        memcpy(work,head,sizeof(head));
        while(int t=dfs(S,INF,T)) ans+=t;
    }
    return ans;
}

int main()
{
    long long sum;
    int t,n,m,S,T,x,y,c,k=1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        S=1;T=n;
        while(m--)
        {
            scanf("%d%d%d",&x,&y,&c);
            add(x,y,c);
        }
        printf("Case %d: %I64d\n",k++,Dinic(S,T));
    }
    return 0;
}