POJ 1459 Power Network（Dinic邻接表+当前弧优化）

题目链接：http://poj.org/problem?id=1459

题意：一个电网包含一些结点（电站、消费者、调度站），这些结点通过电线连接。每个结点u 可能被供给s(u)的电能，s(u)≥0；同时也可能产生p(u)的电能，0≤p(u)≤pmax(u)；站点u 还有可能消费c(u)电能，0≤c(u)≤min( s(u), cmax(u) )；可能传输d(u)的电能，d(u) = s(u) + p(u) - c(u)。以上这些量存在以下限制关系：对每个电站，c(u) = 0；对每个消费者，p(u) = 0；对每个调度站，p(u) = c(u) = 0。

在电网中两个结点u 和v 之间最多有一条电线连接。从结点u 到结点v 传输L(u,v)的电能，0≤L(u,v)≤Lmax(u,v)。定义Con 为c(u)的总和，表示电网中消费电能的总和。本题的目的是求Con 的最大值。

<-摘自《图论算法理论、实现及应用》->

题解：

添加一个超级源点和一个超级汇点，在源点和每个电站之间链接一条容量为pmax的弧，在消费之和汇点之间连接一条容量为cmax的弧。对于题目中给的三元组(u,v)z，从顶点u 连一条容量为z 的弧到顶点v。然后求最大流。

Dinic算法，点太多不能存邻接矩阵，采用一个数组存边，然后用当前弧优化，我把这个优化去掉就超时了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int MAXN=100+10;
const int MAXM=MAXN*MAXN*2;
const int INF=0x3f3f3f3f;
struct arc
{
	int u,v,cap;
	arc(){};
	arc(int u,int v,int cap):u(u),v(v),cap(cap){};
};
arc Edge[MAXM];//存储边的信息
vector<int> pos[MAXN];//存储每个点的邻接边在Edge数组中的位置
int level[MAXN];//层次网络
int current[MAXN];//当前弧
int cnt,N,S,E;

void addEdge(int u,int v,int cap)
{
	pos[u].push_back(cnt);
	Edge[cnt++]=arc(u,v,cap);
	pos[v].push_back(cnt);
	Edge[cnt++]=arc(v,u,0);
}

int BFS()
{
	queue<int> que;
	que.push(S);
	memset(level,-1,sizeof(level));
	level[S]=0;
	while(!que.empty())
	{
		int u=que.front();
		que.pop();
		for(int v=0;v<pos[u].size();v++)
		{
			arc &e=Edge[pos[u][v]];
			if(e.cap>0&&level[e.v]<0)//有残量而且未分配层次
			{
				level[e.v]=level[u]+1;
				que.push(e.v);
			}
		}
	}
	if(level[E]>0) return 1;
	return 0;
}

int DFS(int x,int low)
{
	if(x==E)
		return low;
	for(int i=current[x];i<pos[x].size();i++)
	{
		current[x]=i;//当前弧
		arc &e=Edge[pos[x][i]];//引用
		int a=0;
		if(level[e.v]==level[x]+1&&e.cap>0&&(a=DFS(e.v,min(low,e.cap))))
		{
			e.cap-=a;
			Edge[pos[x][i]^1].cap+=a;
			return a;
		}
	}
	return 0;
}

int Dinic()
{
	int ans=0;
	while(BFS())
	{
		memset(current,0,sizeof(current));
		int tans;
		while(tans=DFS(S,INF)) ans+=tans;
	}
	return ans;
}

int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int n,np,nc,m;
	while(scanf("%d%d%d%d",&n,&np,&nc,&m)!=EOF)
	{
		N=n+2;S=N-2;E=N-1;
		cnt=0;
		for(int i=0;i<N;i++)
		{
			pos[i].clear();
		}
		for(int i=0;i<n;i++)
		{
			pos[i].clear();
		}
		for(int i=0;i<m;i++)
		{
			int u,v,z;
			scanf(" (%d,%d)%d",&u,&v,&z);
			addEdge(u,v,z);
		}
		for(int i=0;i<np;i++)
		{
			int u,z;
			scanf(" (%d)%d",&u,&z);
			addEdge(S,u,z);
		}
		for(int i=0;i<nc;i++)
		{
			int u,z;
			scanf(" (%d)%d",&u,&z);
			addEdge(u,E,z);
		}
	/*	for(int i=0;i<cnt;i++)
		{
			cout<<Edge[i].u<<" "<<Edge[i].v<<" "<<Edge[i].cap<<endl;
		}
	*/	cout<<Dinic()<<endl;
	}
	return 0;
}