网络流学习:(最大流)POJ1149

题意:养猪场有M个猪圈,每个猪圈可以容纳的猪的数量没有限制.养猪场的工人没有钥匙,但是要买猪的客户有若干个猪圈的钥匙.当一个客户来买猪的时候他会把所有可以打开的猪圈打开,客户离开后工人可以将猪圈里剩余的猪分配到任意猪圈.工人预先知道客户的需求和所拥有的钥匙.问养猪场一天最多可以卖出多少头猪.

网络流的题目建图果然恶心….

建图:建立一个源点和一个汇点,如果一个客户i首先打开一个猪圈,将i与源点相连,容量为该猪圈的猪的数量;i后面紧接着有下一个客户j打开同一个猪圈的话,将i与j相连,容量为inf.每一个客户都与汇点相连,容量为该客户的需求.

建好图,直接用Dinic跑一遍就A了.刚看完Dinic算法,印象还是挺深的,没看模板写出来了.

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <queue>
#define MAX 1500
#define inf 10000000
using namespace std;
int level[MAX];//层次网络
int Map[MAX][MAX];//容量网络
bool vis[MAX];//判断猪圈是否被打开过
int N,M;

void Build_Map()
{
int i,j,k,B;
int Pnum[MAX];
memset(vis,0,sizeof(vis));
memset(Map,0,sizeof(Map));
scanf("%d%d",&M,&N);
for(i=1;i<=M;i++)
{
scanf("%d",&Pnum[i]);
}
for(i=1;i<=N;i++)
{
scanf("%d",&k);
for(j=0;j<k;j++)
{
scanf("%d",&B);
if(!vis[B])
{
Map[0][i]+=Pnum[B];
Pnum[B]=i;
vis[B]=1;
}else
{
Map[Pnum[B]][i]=inf;
}
}
scanf("%d",&B);
Map[i][N+1]=B;
}
}

bool BFS()
{
queue<int> q;
memset(level,0,sizeof(level));
level[0]=1;
q.push(0);
while(!q.empty())
{
int v=q.front();
q.pop();
for(int i=0;i<=N+1;i++)
{
if(Map[v][i]&&!level[i])
{
level[i]=level[v]+1;
q.push(i);
}
}
}
if(!level[N+1]) return false;
else return true;
}

int DFS(int v,int sum)
{
if(v==N+1) return sum;
int s=sum;
for(int i=0;sum&&i<=N+1;i++)
{
if(Map[v][i]&&level[v]+1==level[i])
{
int t=DFS(i,min(sum,Map[v][i]));
Map[v][i]-=t;
Map[i][v]+=t;
sum-=t;
}
}
return s-sum;
}

int main()
{
Build_Map();
int ans=0;
while(BFS())
ans+=DFS(0,inf);
printf("%d\n",ans);
return 0;
}