将人拆点,之间连接一条容量为1的边,表示一个人只能选一次,从s到每种食物连接一条容量为该种食物总数的边,从每种饮料连接一条容量为该种饮料总量的边,如果一个人喜欢某种食物或饮料,连边即可。
#include <iostream> #include <cstdio> #include <cstring> using namespace std; const int maxn=1200; const int inf=1000000; int n,f,d; int level[maxn],que[maxn]; int head[maxn],lon; int min(int a,int b) { if(a<b) return a; else return b; } struct { int next,to,c; }e[maxn*1000]; void edgeini() { memset(head,-1,sizeof(head)); lon=-1; } void edgemake(int from,int to,int c) { e[++lon].c=c; e[lon].to=to; e[lon].next=head[from]; head[from]=lon; } void make(int from,int to,int c) { edgemake(from,to,c); edgemake(to,from,0); } bool makelevel(int s,int t) { memset(level,0,sizeof(level)); int front=1,end=0; que[++end]=s; level[s]=1; while(front<=end) { int u=que[front++]; if(u==t) return true; for(int k=head[u];k!=-1;k=e[k].next) { int v=e[k].to; if(!level[v]&&e[k].c) { que[++end]=v; level[v]=level[u]+1; } } } return false; } int dfs(int now,int t,int maxf) { if(now==t) return maxf; int ret=0; for(int k=head[now];k!=-1;k=e[k].next) { int u=e[k].to; if(level[u]==level[now]+1&&e[k].c) { int f=dfs(u,t,min(e[k].c,maxf-ret)); e[k].c-=f; e[k^1].c+=f; ret+=f; if(ret==maxf) return ret; } } if(ret==0) level[now]=0; return ret; } int maxflow(int s,int t) { int ret=0; while(makelevel(s,t)) { ret+=dfs(s,t,inf); } return ret; } int main() { int i,j; while(scanf("%d%d%d",&n,&f,&d)!=EOF) { edgeini(); int t=n+n+f+d+1; int x; char y; for(i=1;i<=f;i++) { scanf("%d",&x); make(0,i,x); } for(i=1;i<=n;i++) make(f+i,f+n+i,1); for(i=1;i<=d;i++) { scanf("%d",&x); make(f+n+n+i,t,x); } getchar(); for(i=1;i<=n;i++) { for(j=1;j<=f;j++) { scanf("%c",&y); if(y=='Y') make(j,i+f,1); } getchar(); } for(i=1;i<=n;i++) { for(j=1;j<=d;j++) { scanf("%c",&y); if(y=='Y') make(i+f+n,j+f+n+n,1); } getchar(); } int ans=maxflow(0,t); printf("%d ",ans); } return 0; }