题面传送门
一道二分图的题目。
考虑将菜向客人,客人向房间连权值为 1 1 1的边,但是这样会有重复计算。
所以将客人拆成两个点,两点之间连权值为 1 1 1的边,就可以达到限制流量的目的。
代码实现:
#include
#include
#include
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,x,y,z,st,t,nows,cur,now[1039],d[1039],ans;
struct yyy{
int to,w,z;}tmp;
struct ljb{
int head,h[1039];
yyy f[10039];
inline void add(int x,int y,int z){
f[head]=(yyy){
y,z,h[x]};
h[x]=head++;
}
}s;
queue<int > q;
inline int bfs(){
while(!q.empty()) q.pop();
memset(d,0x3f,sizeof(d));
d[st]=0;
q.push(st);
now[st]=s.h[st];
while(!q.empty()){
nows=q.front();
q.pop();
cur=s.h[nows];
while(cur!=-1){
tmp=s.f[cur];
if(tmp.w&&d[tmp.to]>=1e9){
d[tmp.to]=d[nows]+1;q.push(tmp.to);now[tmp.to]=s.h[tmp.to];
if(tmp.to==t) return 1;
}
cur=tmp.z;
}
}
return 0;
}
inline int dfs(int x,int sum){
if(x==t) return sum;
int cur=s.h[x],k,pus=0;
yyy tmp;
while(cur!=-1){
tmp=s.f[cur];
if(tmp.w&&d[tmp.to]==d[x]+1){
k=dfs(tmp.to,min(sum,tmp.w));
if(!k) d[tmp.to]=1e9;
s.f[cur].w-=k;
s.f[cur^1].w+=k;
sum-=k;
pus+=k;
}
cur=tmp.z;
if(!sum) break;
}
return pus;
}
int main(){
memset(s.h,-1,sizeof(s.h));
register int i,j;
scanf("%d%d%d",&n,&m,&k);
st=0;t=m+2*n+k+1;
for(i=1;i<=m;i++) s.add(0,i,1),s.add(i,0,0);
for(i=1;i<=n;i++){
for(j=1;j<=m;j++){
scanf("%d",&x);
if(x) s.add(j,i+m,1),s.add(i+m,j,0);
}
}
for(i=1;i<=n;i++) s.add(i+m,i+n+m,1),s.add(i+n+m,i+m,0);
for(i=1;i<=n;i++){
for(j=1;j<=k;j++){
scanf("%d",&x);
if(x) s.add(i+2*n,j+2*n+m,1),s.add(j+2*n+m,i+2*n,0);
}
}
for(i=1;i<=k;i++) s.add(i+2*n+m,t,1),s.add(t,i+2*n+m,0);
while(bfs()) ans+=dfs(st,1e8);
printf("%d\n",ans);
}