题意:
一个农夫养了n头猪,每头猪有自己喜欢的食物和饮料,现在农夫有f中不同的食物,d种不同的饮料,每种食物或者饮料只能分配给一头猪,要求最多能满足多少头猪的要求。
最大流:
这个题的建图很巧妙,首先,我们为了控制一头猪只连一瓶饮料,一份食物,那么我们可以把一头猪拆成两头猪,他们之间连一条权值为1的边,另外左边连它喜欢的食物,权值为1,右边连它喜欢的饮料,权值为1,跑一遍最大流即可。
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<cmath>
#define maxn 600
using namespace std;
int pre[maxn],queue[maxn],map[maxn][maxn],flow[maxn];
int n,f,d,start,end,ans,tot,nf,nd,food,drink;
int bfs()
{
int p=0,q=1;
queue[1]=0;
flow[start]=100000000;
memset(pre,-1,sizeof(pre));
while(p<q)
{
int s=queue[++p];
for(int i=1;i<=end;i++)
{
if(pre[i]==-1&&map[s][i]>0)
{
pre[i]=s;
flow[i]=min(flow[s],map[s][i]);
queue[++q]=i;
if(i==end) return flow[end];
}
}
}
return -1;
}
int Edmonds_Karp()
{
int p;
while((ans=bfs())!=-1)
{
tot+=ans;
int t=end;
while(t!=start)
{
p=pre[t];
map[p][t]-=ans;
map[t][p]+=ans;
t=pre[t];
}
}
return tot;
}
int main()
{
while(scanf("%d%d%d",&n,&f,&d)!=EOF)
{
start=0,end=f+n+n+d+1;
memset(map,0,sizeof(map));
for(int i=1;i<=n;i++)
{
scanf("%d%d",&nf,&nd);
while(nf--)
{
scanf("%d",&food);
map[food][f+i]=1;
}
map[f+i][f+n+i]=1;
while(nd--)
{
scanf("%d",&drink);
map[f+n+i][f+n+n+drink]=1;
}
}
for(int i=1;i<=f;i++)
map[0][i]=1;
for(int i=1;i<=d;i++)
map[f+n+n+i][f+n+n+d+1]=1;
printf("%d\n",Edmonds_Karp());
}
return 0;
}
poj 3281 Dining
最新推荐文章于 2020-06-13 19:48:37 发布