POJ 1273 Drainage Ditches (最大流模板题)

版权声明：欢迎神犇指教 https://blog.csdn.net/sdxtcqs/article/details/82351417

http://poj.org/problem?id=1273
题意 ：现在有$n$个池塘(从1到$n$开始编号，1为源点，$n$为汇点)，及$m$条水渠，给出这$m$条水渠所连接的点和所能流过的最大流量，求从1到$n$能流过的最大流量。

最大流裸题，仿佛学会了dinic算法，就先写这一个版本的吧，这是dinic算法的参考博客，写的炒鸡好https://www.cnblogs.com/SYCstudio/p/7260613.html。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f
using namespace std;

const int maxn=100100;
int head[maxn],dis[maxn],k,n,m;
struct edge{
    int v,c,next;
}e[maxn];
void addedge(int u,int v,int c)
{
    e[k].v=v;e[k].c=c;e[k].next=head[u];head[u]=k++;
    e[k].v=u;e[k].c=0;e[k].next=head[v];head[v]=k++;//建图时连容量为0的反向边
}

int bfs(int s,int t)//bfs出每一个点到源点距离的层次图
{
    memset(dis,0,sizeof(dis));
    dis[s]=1;
    queue<int> q;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            int v=e[i].v;
            if(!dis[v]&&e[i].c)
            {
                dis[v]=dis[u]+1;
                if(v==t) return 1;
                q.push(v);
            }
        }
    }
    return 0;
}

int dfs(int u,int a,int t)//u是当前节点，a是当前流量，t是汇点
{
    if(u==t) return a;
    int res=0;
    for(int& i=cur[u];i!=-1;i=e[i].next)//当前弧优化，这里直接定义成引用，i++的时候cur[u]的值也同时改变
    {
        if(!a) break;
        int v=e[i].v;
        if(e[i].c&&dis[v]==dis[u]+1)//满足残量不为0和层次图距离大1两个条件
        {
            int f=dfs(v,min(a,e[i].c),t);//向下增广
            if(f>0)//増广成功
            {
                e[i].c-=f;//正向边减
                e[i^1].c+=f;//反向边加
                res+=f;
                a-=f;
            }
        }
    }
    return res;
}

int dinic(int s,int t)
{
    int res=0;
    while(bfs(s,t))
    {
        for(int i=1;i<=n;i++)//每次dfs前都要将cur[i]赋值为head[i]
            cur[i]=head[i];
        res+=dfs(s,INF,t);
    }
    return res;
}

int main()
{
    int x,y,z;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        memset(e,0,sizeof(e));
        memset(head,-1,sizeof(head));
        k=0;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            addedge(x,y,z);
        }
        printf("%d\n",dinic(1,n));
    }
    return 0;
}