POJ 1149 PIGS

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

网络流，主要是建图的问题

题意:

有 M 个猪圈（M ≤ 1000），每个猪圈里初始时有若干头猪。

一开始所有猪圈都是关闭的。

依次来了 N 个顾客（N ≤ 100），每个顾客分别会打开指定的几个猪圈，从中买若干头猪。

每个顾客分别都有他能够买的数量的上限。

每个顾客走后，他打开的那些猪圈中的猪，都可以被任意地调换到其它开着的猪圈里，然后所有猪圈重新关上。

问总共最多能卖出多少头猪。

注意： 顾客是一个接一个的来买猪，而不是同时来

 

建图：

每个顾客分别用一个节点来表示。

对于每个猪圈的第一个顾客，从源点向他连一条边，容量就是该猪圈里的猪的初始数量。如果从源点到一名顾客有多条边，则可以把它们合并成一条，容量相加。

若第i个人与他后面的第j个人有同一个猪圈的钥匙，则从i引边到j,容量为无穷大

从各个顾客到汇点各有一条边，容量是各个顾客能买的数量上限

#include<stdio.h>
#include<string.h>
#include<vector>
#define M 120
#define INF 0xfffffff
using namespace std;

int n,m;
int g[M][M];
int d[M];
int stack[M];
int que[M*M];
int num[1001];//各个猪圈猪的数量 
vector<int> G[M];//图的邻接表 
vector<int> cus[1001];//cus[i]保存的是能打开i号猪圈的所有顾客 

bool BFS(int s,int t)
{
	int l=0,r=0;
	int i,p;
	memset(d,-1,sizeof(d));
	d[s]=0;
	que[r++]=s;
	while(l<r)
	{
		p=que[l++];
		for(i=0;i<G[p].size();i++)
		{
			int u=G[p][i];
			if( g[p][u]>0 && d[u]==-1)
			{
				d[u]=d[p]+1;
				if(u==t) 
					return true;
				que[r++]=u;
			}
		}
	}
	return false;
}

int Dinic(int s,int t)
{
	int i,p;
	int mn;
	int nMinc;
	int maxflow=0;
	bool v[M];
	
	while(BFS(s,t))
	{
		int top=0,base=0;
		stack[top++]=s;
		memset(v,false,sizeof(v));
		v[s]=true;
		while(top>base)
		{
			p=stack[top-1];
			if(p==t)
			{
				mn=INF;
				for(i=1;i<top;i++)
				{
					int u=stack[i-1];
					int v=stack[i];
					if(g[u][v]<mn)
					{
						mn=g[u][v];
						nMinc=u;
					}
				}
				maxflow+=mn;
				for(i=1;i<top;i++)
				{
					int u=stack[i-1];
					int v=stack[i];
					g[u][v]-=mn;
					g[v][u]+=mn;
				}
				while(top>base && stack[top-1]!=nMinc)
					top--;
			}
			else 
			{
				for(i=0;i<G[p].size();i++)
				{
					int u=G[p][i];
					if( !v[u] && g[p][u]>0 && d[u]==d[p]+1)
					{
						v[u]=true;
						stack[top++]=u;
						break;
					}
				}
				if(i==G[p].size())
					top--;
			}
		}
	}
	return maxflow;
}

int main()
{
	int i,j;
	while(scanf("%d%d",&m,&n)!=EOF)
	{
		memset(g,0,sizeof(g));
		memset(num,0,sizeof(num));
		memset(cus,0,sizeof(cus));
		memset(G,0,sizeof(G));
		for(i=1;i<=m;i++)
			scanf("%d",&num[i]);
		for(i=1;i<=n;i++)
		{
			int k,p;
			scanf("%d",&k);
			while(k--)
			{
				int a;
				scanf("%d",&a);
				if(cus[a].size()==0)//如果i是第一个能打开a猪圈的顾客
				{	
					g[0][i]+=num[a];
					cus[a].push_back(i);
					G[0].push_back(i);
				}
				else 
				{
					for(j=0;j<cus[a].size();j++)//每个能打开a猪圈的顾客都向i连一条无穷的的边 
					{
						int u=cus[a][j];
						g[u][i]=INF;
						G[u].push_back(i);
						G[i].push_back(u);
					}
				}
			}
			scanf("%d",&p);
			g[i][n+1]=p;//每个顾客与汇连一条边 
			G[i].push_back(n+1);
		}
		printf("%d\n",Dinic(0,n+1));
	}
	return 0;
}