题目地址:HDU 4292
水题。
由于每个人只能有1份,所以需要拆点限制流量。建图方法为,建一源点与汇点,将食物与源点相连,权值为食物额数量,将饮料与汇点相连,权值为饮料数量。。然后将人进行拆点为i和i‘,将对应的i与i’连边权值为1,将i与它所对应的YES的食物连边,将i‘与它所对应的YES的饮料连边,一次求最大流。
代码如下:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
int head[900], source, sink, nv, cnt;
int cur[900], num[900], d[900], pre[900], q[900];
struct node
{
int u, v, cap, next;
} edge[10000000];
void add(int u, int v, int cap)
{
edge[cnt].v=v;
edge[cnt].cap=cap;
edge[cnt].next=head[u];
head[u]=cnt++;
edge[cnt].v=u;
edge[cnt].cap=0;
edge[cnt].next=head[v];
head[v]=cnt++;
}
void bfs()
{
memset(num,0,sizeof(num));
memset(d,-1,sizeof(d));
int f1=0, f2=0, i;
d[sink]=0;
num[0]=1;
q[f1++]=sink;
while(f1>=f2)
{
int u=q[f2++];
for(i=head[u]; i!=-1; i=edge[i].next)
{
int v=edge[i].v;
if(d[v]==-1)
{
d[v]=d[u]+1;
num[d[v]]++;
q[f1++]=v;
}
}
}
}
void isap()
{
memcpy(cur,head,sizeof(cur));
bfs();
int flow=0, u=pre[source]=source, i;
while(d[source]<nv)
{
if(u==sink)
{
int f=INF,pos;
for(i=source; i!=sink; i=edge[cur[i]].v)
{
if(f>edge[cur[i]].cap)
{
f=edge[cur[i]].cap;
pos=i;
}
}
for(i=source; i!=sink; i=edge[cur[i]].v)
{
edge[cur[i]].cap-=f;
edge[cur[i]^1].cap+=f;
}
flow+=f;
u=pos;
}
for(i=cur[u]; i!=-1; i=edge[i].next)
{
if(d[edge[i].v]+1==d[u]&&edge[i].cap)
{
break;
}
}
if(i!=-1)
{
cur[u]=i;
pre[edge[i].v]=u;
u=edge[i].v;
}
else
{
if(--num[d[u]]==0) break;
int mind=nv;
for(i=head[u]; i!=-1; i=edge[i].next)
{
if(mind>d[edge[i].v]&&edge[i].cap)
{
mind=d[edge[i].v];
cur[u]=i;
}
}
d[u]=mind+1;
num[d[u]]++;
u=pre[u];
}
}
printf("%d\n",flow);
}
int main()
{
int n, f, d, i, j, x;
char s[300];
while(scanf("%d%d%d",&n,&f,&d)!=EOF)
{
memset(head,-1,sizeof(head));
source=0;
cnt=0;
sink=2*n+f+d+1;
nv=sink+1;
for(i=1; i<=f; i++)
{
scanf("%d",&x);
add(source,i,x);
}
for(i=1; i<=d; i++)
{
scanf("%d",&x);
add(i+2*n+f,sink,x);
}
for(i=1; i<=n; i++)
{
add(i+f,i+f+n,1);
}
for(i=1; i<=n; i++)
{
scanf("%s",s);
for(j=0; j<f; j++)
{
if(s[j]=='Y')
{
add(j+1,f+i,INF);
}
}
}
for(i=1; i<=n; i++)
{
scanf("%s",s);
for(j=0; j<d; j++)
{
if(s[j]=='Y')
{
add(i+n+f,j+1+2*n+f,INF);
}
}
}
isap();
}
return 0;
}