圆桌问题
考虑建图,设置源点S为0,汇点T为n+m+1,源点向所有单位连其容量的边,每个单位向圆桌连1的边,每个圆桌向源点连圆桌的容量的边,
可知符合网络流的容量限制和流量限制,其最大流就是合法方案之一。
#include<bits/stdc++.h>
using namespace std;
const int N=2000,M=80000,INF=0x3f3f3f3f;
int n,m,S,T;
int h[N],ne[M],e[M],f[M],idx;
int cur[N],d[N];
int r[N];
int q[N];
int c[N];
void add(int a,int b,int c)
{
e[idx]=b,ne[idx]=h[a],f[idx]=c,h[a]=idx++;
}
bool bfs()
{
memset(d,-1,sizeof(d));
int hh=-1,tt=0;
q[++hh]=S;
cur[S]=h[S];
d[S]=0;
while(hh<=tt)
{
int t=q[hh++];
for(int i=h[t];~i;i=ne[i])
{
int ver=e[i];
if(d[ver]==-1&&f[i])
{
d[ver]=d[t]+1;
cur[ver]=h[ver];
if(ver==T) return true;
q[++tt]=ver;
}
}
}
return false;
}
int find(int u,int limit)
{
if(u==T) return limit;
int flow=0;
for(int i=cur[u];~i&&flow<limit;i=ne[i])
{
cur[u]=i;
int ver=e[i];
if(d[ver]==d[u]+1&&f[i])
{
int t=find(ver,min(f[i],limit-flow));
if(!t) d[ver]=-1;
f[i]-=t,f[i^1]+=t,flow+=t;
}
}
return flow;
}
int dinic()
{
int r=0;
int flow;
while(bfs()) while(flow=find(S,INF)) r+=flow;
return r;
}
int main()
{
cin>>m>>n;
int sum=0;
memset(h,-1,sizeof(h));
for(int i=1;i<=m;i++)
{
cin>>r[i];
sum+=r[i];
}
for(int i=1;i<=n;i++)
{
cin>>c[i];
}
S=0,T=n+m+1;
for(int i=1;i<=m;i++)
{
add(S,i,r[i]);
add(i,S,0);
}
for(int i=1;i<=m;i++)
{
for(int j=m+1;j<=n+m;j++)
{
add(i,j,1);
add(j,i,0);
}
}
for(int i=m+1;i<=n+m;i++)
{
add(i,T,c[i-m]);
add(T,i,0);
}
if((dinic())==sum)
{
cout<<1<<endl;
for(int i=1;i<=m;i++)
{
for(int j=h[i];~j;j=ne[j])
{
if(e[j]>m&&e[j]<n+m+1&&!f[j])
{
cout<<e[j]-m<<' ';
}
}
cout<<"\n";
}
}else
{
cout<<0<<"\n";
}
}