POJ 1273 Drainage Ditches

网络流-最大流入门题。

 

CODE：

 

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
using  namespace std;

const  int M =  201;
const  int INF =  0x3f3f3f3f;
int flow[M][M], cap[M][M];
int a[M], p[M];
int s, t;
int n, m;

void init()
{
    memset(p,  0,  sizeof(p));
    memset(cap,  0,  sizeof(cap));
}

int EK( int s,  int t)
{
    queue< int> q;
    memset(flow,  0,  sizeof(flow));
     int f =  0;
     for(;;)
    {
        memset(a,  0,  sizeof(a));              //a[i]保存s到每个节点i路径上的最小参量，a[t]就是整条s-t道路上的最小残量。
        a[s] = INF;
        q.push(s);
         while(!q.empty())   //BFS找增广路径
        {
             int u = q.front(); q.pop();         //记录v的父亲，并加入FIFO队列。
             for( int v =  1; v <= n; v++)  if(!a[v] && cap[u][v]>flow[u][v])
            {
                p[v] = u; q.push(v);      //记录v的父亲，并加入FIFO队列
                a[v] = min(a[u], cap[u][v]-flow[u][v]); //s-v路径上的最小残量
            }
        }
         if(a[t] ==  0)  break;
         for( int u = t; u != s; u = p[u])
        {
            flow[p[u]][u] += a[t];         //更新正向流量
            flow[u][p[u]] -= a[t];         //更新反向流量
        }
        f += a[t];
    }
     return f;
}

int main()
{
     while(~scanf( " %d%d ", &m, &n))
    {
        init();
         while(m--)
        {
             int u, v, w;
            scanf( " %d%d%d ", &u, &v, &w);
            cap[u][v] += w;
        }
         int ans = EK( 1, n);
        printf( " %d\n ", ans);
    }
     return  0;
}