(网络流24题大多需要spj,所以需要一个有spj的oj,本系列代码均在www.oj.swust.edu.cn测试通过)
这道题的模型应该是有一种专业名称的,但是那都没用,思路才是最关键的。
我们从源点向每一个公司连接一条容量为公司人数的边,然后每个公司向所有餐桌连一条容量为1得边,每个餐桌向汇点连一条容量为餐桌人数上限的边,这样跑一边最大流,如果最大流等于所有公司的总人数之和,证明有解,反之无解
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<cstring>
#include<string>
#include<iomanip>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF 100000000
struct bian
{
int l,r,f,lei;
}a[1000000];
int s=0,t=99999;
int d[100000];
int fir[1000000];
int nex[1000000];
int wz[1000000];
bool bfs()
{
static int dui[100000];
memset(d,-1,sizeof(d));
int top=1,my_final=2;
dui[1]=s;
d[s]=1;
while(top<my_final)
{
int u=dui[top++];
for(int o=fir[u];o;o=nex[o])
{
if(a[o].f && d[a[o].r]==-1)
{
d[a[o].r]=d[u]+1;
dui[my_final++]=a[o].r;
if(a[o].r==t) return true;
}
}
}
return false;
}
int dinic(int u,int flow)
{
if(u==t) return flow;
int left=flow;
for(int o=fir[u];o&&left;o=nex[o])
{
if(a[o].f && d[a[o].r]==d[u]+1)
{
int temp=dinic(a[o].r,min(a[o].f,left));
if(!temp) d[a[o].r]=-1;
left-=temp;
a[o].f-=temp;
a[o^1].f+=temp;
}
}
return flow-left;
}
int tot=1;
void add_edge(int l,int r,int f,int lei=0)
{
a[++tot].l=l;
a[tot].r=r;
a[tot].f=f;
a[tot].lei=lei;
nex[tot]=fir[l];
fir[l]=tot;
}
int main()
{
int n,m;
scanf("%d%d",&m,&n);
int zong=0;
for(int i=1;i<=m;i++)
{
int x;
scanf("%d",&x);
zong+=x;
add_edge(s,i,x);
add_edge(i,s,0);
}
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
add_edge(m+i,t,x);
add_edge(t,m+i,0);
}
for(int i=1;i<=m;i++)
{
wz[i]=tot+1;
for(int j=1;j<=n;j++)
{
add_edge(i,m+j,1,1);
add_edge(m+j,i,0);
}
}
wz[m+1]=tot+1;
int ans=0;
while(bfs()) ans+=dinic(s,INF);
if(ans!=zong) {cout<<0;return 0;}
else
{
cout<<1<<endl;
for(int i=1;i<=m;i++)
{
for(int j=wz[i];j<wz[i+1];j++) if(a[j].f==0 && a[j].lei) printf("%d ",a[j].r-m);
printf("\n");
}
}
return 0;
}