POJ 1459 Power Network(最大流)

POJ 1459 Power Network(最大流)

http://poj.org/problem?id=1459

题意:

       (原题描述的过于复杂,原题中的s[i]根本不用管)总共有n个节点，其中有发电站np个、用户nc个和调度器n-np-nc个三种节点以及M条电线(用于传输电流的)，每个发电站有一个最大发电量，每个用户有个最大接受电量，现在有m条有向边，边有一个最大的电流量，表示最多可以流出这么多电，现在从发电站发电到用户，问最多可以发多少电(被用户接受).

分析:

       建立一个超级源点s,s到任意发电站i有(s,i,,p[i]) (表示该发电站最多能发p[i]电)

       建立一个超级汇点t,任意用户j到t有边(j,t,c[j]) (表示该用户最多能消费c[j]电)

       然后对于题目中描述的M条电线(u,v)L, 就有边(u,v,L).

       最终我们求本图的最大流即可.(到汇点t的最大流就是所有用户能消费的最大电量)

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define INF 1e9
using namespace std;
const int maxn=100+5;

struct Edge
{
    int from,to,cap,flow;
    Edge(){}
    Edge(int f,int t,int c,int fl):from(f),to(t),cap(c),flow(fl){}
};

struct Dinic
{
    int n,m,s,t;
    vector<Edge> edges;
    vector<int> G[maxn];
    int cur[maxn];
    bool vis[maxn];
    int d[maxn];

    void init(int n,int s,int t)
    {
        this->n=n, this->s=s, this->t=t;
        edges.clear();
        for(int i=0;i<n;i++) G[i].clear();
    }

    void AddEdge(int from,int to,int cap)
    {
        edges.push_back( Edge(from,to,cap,0) );
        edges.push_back( Edge(to,from,0,0) );
        m=edges.size();
        G[from].push_back( m-2 );
        G[to].push_back(m-1);
    }

    bool BFS()
    {
        queue<int> Q;
        memset(vis,0,sizeof(vis));
        vis[s]=true;
        d[s]=0;
        Q.push(s);
        while(!Q.empty())
        {
            int x= Q.front(); Q.pop();
            for(int i=0;i<G[x].size();++i)
            {
                Edge& e=edges[G[x][i]];
                if(!vis[e.to] && e.cap>e.flow)
                {
                    vis[e.to]=true;
                    d[e.to]=d[x]+1;
                    Q.push(e.to);
                }
            }
        }
        return vis[t];
    }

    int DFS(int x,int a)
    {
        if(x==t || a==0) return a;
        int flow=0,f;
        for(int& i=cur[x];i<G[x].size();i++)
        {
            Edge& e=edges[G[x][i]];
            if(d[e.to]==d[x]+1 && (f=DFS(e.to,min(a,e.cap-e.flow) ) )>0)
            {
                e.flow +=f;
                edges[G[x][i]^1].flow -=f;
                flow +=f;
                a-=f;
                if(a==0) break;
            }
        }
        return flow;
    }

    int max_flow()
    {
        int ans=0;
        while(BFS())
        {
            memset(cur,0,sizeof(cur));
            ans+= DFS(s,INF);
        }
        return ans;
    }
}DC;

int main()
{
    int n,np,nc,m;
    while(scanf("%d%d%d%d",&n,&np,&nc,&m)==4)
    {
        DC.init(n+2,0,n+1);
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            scanf(" (%d,%d)%d",&u,&v,&w);
            ++u,++v;
            DC.AddEdge(u,v,w);
        }
        for(int i=0;i<np;i++)
        {
            int u,w;
            scanf(" (%d)%d",&u,&w);
            ++u;
            DC.AddEdge(0,u,w);
        }
        for(int i=0;i<nc;i++)
        {
            int u,w;
            scanf(" (%d)%d",&u,&w);
            ++u;
            DC.AddEdge(u,n+1,w);
        }
        printf("%d\n",DC.max_flow());
    }
    return 0;
}