题意:有n个人,每个人都有自己喜欢的食物和饮料,如果一个人不能得到自己的喜欢的,则他将离开,问如何分配可以使剩下的人最多
思路:刚刚写的时候还以为是二分图匹配呢,无情wa,之前写过POJ的3281,一个类型嘛,没有区别,改了之后A之,这里解释一下为什么牛会和牛在连一条容量为1的边,这是因为如果一个牛可以匹配好多食物和饮料,如果不限制的话,这一个牛就可以匹配多种食物,但是我们要的是最多的牛的数量,加了这条边后每个牛就只能走一次了
#include <queue> #include <vector> #include <stdio.h> #include <string.h> #include <stdlib.h> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; const int inf=0x3f3f3f3f; const int maxn=10010; struct edge{ int to,cap,rev; edge(){} edge(int a,int b,int c){to=a;cap=b;rev=c;} }; vector<edge> G[maxn]; int level[maxn],iter[maxn]; void add_edge(int from,int to,int cap){ G[from].push_back(edge(to,cap,G[to].size())); G[to].push_back(edge(from,0,G[from].size()-1)); } void bfs(int s){ memset(level,-1,sizeof(level)); queue<int>que; level[s]=0;que.push(s); while(!que.empty()){ int v=que.front();que.pop(); for(unsigned int i=0;i<G[v].size();i++){ edge &e=G[v][i]; if(e.cap>0&&level[e.to]<0){ level[e.to]=level[v]+1; que.push(e.to); } } } } int dfs(int v,int t,int f){ if(v==t) return f; for(int &i=iter[v];i<G[v].size();i++){ edge &e=G[v][i]; if(e.cap>0&&level[v]<level[e.to]){ int d=dfs(e.to,t,min(f,e.cap)); if(d>0){ e.cap-=d; G[e.to][e.rev].cap+=d; return d; } } } return 0; } int max_flow(int s,int t){ int flow=0; while(1){ bfs(s); if(level[t]<0) return flow; memset(iter,0,sizeof(iter)); int f; while((f=dfs(s,t,inf))>0) flow+=f; } } int main(){ int n,f,d,a,b,c; char str[1010]; while(scanf("%d%d%d",&n,&f,&d)!=EOF){ for(int i=0;i<maxn;i++) G[i].clear(); for(int i=1;i<=f;i++){ scanf("%d",&a); add_edge(0,i,a); } for(int i=1;i<=d;i++){ scanf("%d",&a); add_edge(f+n+i+n,n+n+f+d+1,a); } for(int i=1;i<=n;i++) add_edge(f+i,f+i+n,1); for(int i=1;i<=n;i++){ scanf("%s",str); for(int j=0;j<f;j++){ if(str[j]=='Y') add_edge(j+1,i+f,1); } } for(int i=1;i<=n;i++){ scanf("%s",str); for(int j=0;j<d;j++){ if(str[j]=='Y') add_edge(i+f+n,j+1+n+f+n,1); } } int ans=max_flow(0,n+n+f+d+1); printf("%d\n",ans); } return 0; }
时间: 2024-10-07 12:20:51