hdu3549 Dinic

边表存储的Dinic模版

#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;

struct node
{
    int u, v, f;
};

node e[M];
int next[M], first[N], cc;
int work[N], dis[N], q[N];

/*
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()
{
    cc = 0;
    memset(first, -1, sizeof(first));
}

void add(int u,int v,int f)  //增加一条从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++;
    */
    e[cc].u = u;
    e[cc].v = v;
    e[cc].f = f;
    next[cc] = first[u];
    first[u] = cc;
    cc++;

    e[cc].v = u;
    e[cc].u = v;
    e[cc].f = 0;
    next[cc] = first[v];
    first[v] = cc;
    cc++;
}


//广搜计算出每个点与源点的最短距离，如果不能到达汇点说明算法结束
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=first[q[i]];j!=-1;j=next[j])
        {
            if( e[j].f && dis[ e[j].v ] == -1 )
            {
                dis[e[j].v]=dis[q[i]]+1;
                q[rear++]=e[j].v;
                if(e[j].v==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(e[i].f&&dis[e[i].v]==dis[cur]+1)
        {
            int t=dfs(e[i].v,min(a,e[i].f),T);
            if(t)
            {
                e[i].f-=t;
                e[i^1].f+=t;  //正反向边容量改变
                return t;
            }
        }
    }
    return 0;
}

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

int main()
{
    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;
}