POJ Power Network (dinic)

题目大意:给定n个点,a个发电站,b个用户,m个线路,求最大流的问题.

思路:要改成单源单汇点来进行处理.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<queue>
#include<math.h>
#include<algorithm>
#define ll __int64
#define inf 0x3f3f3f3f
using namespace std;
int n,s,t,cnt,head[200000],l[200000];

struct node
{
    int to,next,r,w;
}q[200000];

void add(int a,int b,int c)
{
    q[cnt].to=b;
    q[cnt].next=head[a];
    q[cnt].w=c;
    head[a]=cnt++;

    q[cnt].to=a;
    q[cnt].next=head[b];
    q[cnt].r=0;
    head[b]=cnt++;
}
int bfs()
{
    queue<int>Q;
    while(!Q.empty())
        Q.pop();
    memset(l,-1,sizeof(l));
    l[s]=0;
    Q.push(s);
    while(!Q.empty())
    {
        int v=Q.front();
        Q.pop();
        for(int i=head[v];~i;i=q[i].next)
        {
            if(q[i].w&&l[q[i].to ]==-1)
            {
                l[q[i].to ]=l[v]+1;
                Q.push(q[i].to);
            }
        }
    }
    if(l[t]>0) return 1;
    else return 0;
}
int dfs(int x,int f)
{
    int a;
    if(x==t)
        return f;
    for(int i=head[x];~i;i=q[i].next)
    {
        if(q[i].w&& (l[q[i].to ]==l[x]+1 )&&(a=dfs( q[i].to,min(q[i].w,f) )) )
        {
            q[i].w-=a;
            q[q[i].r].w+=a;
            return a;
        }
    }
    l[x]=-1;//没有汇超时
    return 0;
}
int main()
{
    int j,k,m,i,a,b;
    while(~scanf("%d%d%d%d",&n,&a,&b,&m))
    {
        cnt=1;//注意每个点都加了1所以要初始化为1 
        s=0,t=n+1;
        int u,v,w;
        memset(head,-1,sizeof(head));
        while(m--)
        {
            scanf(" (%d,%d)%d",&u,&v,&w);
            add(u+1,v+1,w);
        }
        while(a--)
        {
            scanf(" (%d)%d",&u,&w);
            add(s,u+1,w);
        }
        while(b--)
        {
            scanf(" (%d)%d",&u,&w);
            add(u+1,t,w);
        }
        int ans=0;
        while(bfs())
        {
            while(k=dfs(0,inf))
            {
                ans+=k;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}