【最大流（EK）】poj 1149 PIGS

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

题意：有M个猪圈、N个顾客，每个顾客有A把钥匙；只能在有钥匙的猪圈购买最多B只猪，顾客购买过后，可以对打开的猪圈内的猪重新分配；求最多可以卖出多少头猪？

分析：经过大神的总结，得以建立以下图：

1）在原先的图中添加源点（0）和汇点（n+1）；

2）源点连接的是每个猪圈的第一个顾客，如果一个顾客是多个猪圈的第一名买家，则将权值合并；

3）每个顾客和都和汇点连接，权值为最多可购买猪的数量；

4）如果第j个顾客紧接着在第i个顾客后，打开某猪圈，则他们之间的边(i,j)的权值为无穷大；

可将样例构造成如下图：

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

const int MAX=0xfffffff;
const int NM=105;
int c[1005],a[NM][NM],pre[NM],save[NM],n,m,ans;
bool vis[NM];

bool BFS(int s,int d)
{
	int i,t;
	queue<int>q1;
	memset(vis,0,sizeof(vis));
	memset(pre,0,sizeof(pre));
	
	pre[s]=s;
	vis[s]=1;
	q1.push(s);
	while(!q1.empty())
	{
		t=q1.front();q1.pop();
		for(i=1;i<=n+1;i++){
			if(!vis[i]&&a[t][i]){
				vis[i]=1;
				pre[i]=t;
				if(i==d) return true;
				q1.push(i);
			}
		}
	}
	return false;
}

void EK(int s,int d)
{
	int i,v;
	while(BFS(s,d))
	{
		v=MAX;
		for(i=d;i!=s;i=pre[i]){
			if(v>a[pre[i]][i]) v=a[pre[i]][i];
		}
		for(i=d;i!=s;i=pre[i]){
			a[pre[i]][i]-=v;
			a[i][pre[i]]+=v;
		}
		ans+=v;
	}
}

int main()
{
	int i,j,k,v,t;
	while(scanf("%d%d",&m,&n)!=EOF)
	{
		memset(a,0,sizeof(a));
		for(i=1;i<=m;i++)
			scanf("%d",&c[i]);
		for(i=1;i<=n;i++)
		{
			scanf("%d",&k);
			v=0;
			for(j=0;j<k;j++){
				scanf("%d",&t);
				v+=c[t];
				if(!c[t]) a[save[t]][i]=MAX;  //如果不是第一个顾客
				else c[t]=0;
				save[t]=i; 
			}
			if(v) a[0][i]=v;  //存在是该顾客第一个访问的猪圈
			scanf("%d",&t);
			a[i][n+1]=t;
		}
		ans=0;
		EK(0,n+1);
		printf("%d\n",ans);
	}
	return 0;
}