题目链接:点击进入、
思路
源点->房间->人(拆点)->菜->汇点
代码
#include<cstdio>
#include<algorithm>
#include<queue>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1e6+10;
struct node
{
int to;
int next;
ll val;
}edge[maxn];
int head[maxn],tot;
void add(int u,int v,int val)
{
edge[tot].to=v;
edge[tot].val=val;
edge[tot].next=head[u];
head[u]=tot++;
}
int deep[maxn];
int cur[maxn];
int n,p,q,x;
ll res;
int s;
int t;
int dfs(int pos,int flow)
{
if(pos==t) return flow;
int rest=flow,k,i;
for(i=cur[pos];i!=-1&&rest;i=edge[i].next)
{
int v=edge[i].to;
int val=edge[i].val;
if(deep[v]==deep[pos]+1&&val)
{
k=dfs(v,min(val,rest));
if(!k) deep[v]=0;
edge[i].val-=k;
edge[i^1].val+=k;
rest-=k;
}
}
cur[pos]=i;
return flow-rest;
}
bool bfs()
{
memset(deep,0,sizeof(deep));
deep[s]=1;
queue<int>q;
q.push(s);
cur[s]=head[s];
while(q.size())
{
int now=q.front();
q.pop();
for(int i=head[now];i!=-1;i=edge[i].next)
{
int v=edge[i].to;
if(edge[i].val&&!deep[v])
{
q.push(v);
cur[v]=head[v];
deep[v]=deep[now]+1;
if(v==t) return 1;
}
}
}
return 0;
}
void dinic()
{
while(bfs())
res+=dfs(s,0x3f3f3f3f);
}
int main()
{
ios::sync_with_stdio(false);
memset(head,-1,sizeof(head));
cin>>n>>p>>q;
s=2*n+p+q+1,t=2*n+p+q+2;
for(int i=1;i<=p;i++)
{
add(s,i,1);
add(i,s,0);
}
for(int i=p+2*n+1;i<=p+2*n+q;i++)
{
add(i,t,1);
add(t,i,0);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=p;j++)
{
cin>>x;
if(x==1)
{
add(j,i+p,1);
add(i+p,j,0);
}
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=q;j++)
{
cin>>x;
if(x==1)
{
add(i+p+n,j+p+2*n,1);
add(j+p+2*n,i+p+n,0);
}
}
}
for(int i=1;i<=n;i++)
{
add(i+p,i+p+n,1);
add(i+p+n,i+p,0);
}
dinic();
printf("%lld\n",res);
return 0;
}