题目大意:有n个人,p个房间,q道菜,每个房间只能住一个人,每道菜只能给一个人吃。每个人有一些自己喜欢的房间和一些自己喜欢的菜。给一个人住喜欢的房间和吃喜欢的菜才能让他满意,求最多能让多少人满意。
解题思路:最大流,建立超级源点,向每个房间连容量为1的边,超级汇点,每道菜向超级汇点连接一条容量为1的边,然后每个房间向喜欢这个房间的人连接一条容量为1的边,每个人向自己喜欢的菜连接一条容量为1的边。然后你就WA了(๑ŐдŐ)b
因为这样做的话,一个人就可能住多个房间,吃多道菜,所以需要拆点。拆完发现AC了(^o^)/~
以下是Dinic算法代码。
C++ Code:
#include<cstdio> #include<vector> #include<cstring> #include<queue> using namespace std; struct edge{ int from,to,cap,rev; }; vector<edge>G[220000]; queue<int>Q; int n,p,q,iter[220000],level[220000]; inline void addedge(int from,int to,int cap){ G[from].push_back((edge){from,to,cap,G[to].size()}); G[to].push_back((edge){to,from,0,G[from].size()-1}); } void bfs(int s,int t){ memset(level,-1,sizeof level); level[s]=0; Q.push(s); while(!Q.empty()){ int u=Q.front(); Q.pop(); for(int i=0;i<G[u].size();++i){ edge& e=G[u][i]; if(e.cap>0&&level[e.to]<0){ level[e.to]=level[u]+1; Q.push(e.to); } } } } int dfs(int u,int t,int f){ if(u==t)return f; for(int& i=iter[u];i<G[u].size();++i){ edge& e=G[u][i]; if(e.cap>0&&level[e.to]>level[u]){ int d=dfs(e.to,t,(e.cap>f)?(f):(e.cap)); if(d){ e.cap-=d; G[e.to][e.rev].cap+=d; return d; } } } return 0; } int max$flow(int s,int t){ int flow=0; for(;;){ bfs(s,t); if(level[t]<0)return flow; memset(iter,0,sizeof iter); int f; while(f=dfs(s,t,0x3f3f3f3f))flow+=f; } } int main(){ scanf("%d%d%d",&n,&p,&q); for(int i=1;i<=n;++i) for(int j=1;j<=p;++j){ int x; scanf("%d",&x); if(x){ addedge(n*2+j,i,1); } } for(int i=1;i<=n;++i) for(int j=1;j<=q;++j){ int x; scanf("%d",&x); if(x){ addedge(i+n,n*2+p+j,1); } } for(int i=1;i<=n;++i)addedge(i,i+n,1); for(int i=1;i<=p;++i)addedge(0,n*2+i,1); for(int i=1;i<=q;++i)addedge(n*2+p+i,n*2+p+q+1,1); int ans=max$flow(0,n*2+p+q+1); printf("%d ",ans); return 0; }