给出
m
≤
150
m\leq150
m≤150个单位,
n
≤
270
n\leq270
n≤270个餐桌,并且给出了每个单位的人数以及每个餐桌容纳的人数。求一个满足的方案,不能的话就直接输出
0
0
0。
这个题简单了。就是直接源点流人数的流量到单位,单位流
1
1
1的流量到餐桌,餐桌流餐桌的人数到汇点。判断是不是人数能流完,流完说明有解。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int sum=0;
namespace Dinic {
const int N=1e4+3,M=1e5+3;
int front[N],head[N],dep[N],tot=-1,n,S,T;
int m;
struct edge{ int v,nxt;ll w; }e[2*M];
void init() {
tot=-1;
for(int i=1;i<=n;i++) front[i]=-1;
}
void add(int u,int v,ll w) {
e[++tot].v=v,e[tot].w=w,e[tot].nxt=front[u],front[u]=tot;
e[++tot].v=u,e[tot].w=0,e[tot].nxt=front[v],front[v]=tot;
}
bool bfs() {
for(int i=1;i<=n;i++) dep[i]=0;
queue<int> q;
q.push(S);
dep[S]=1;
while(!q.empty()) {
int u=q.front();
q.pop();
for(int i=front[u];i!=-1;i=e[i].nxt) {
int v=e[i].v;
if(dep[v]==0&&e[i].w) {
dep[v]=dep[u]+1;
q.push(v);
}
}
}
return dep[T];
}
ll dfs(int cur,ll dis=1e18) {
if(cur==T) return dis;
ll flow=0,sum=0;
for(int &i=head[cur];i!=-1;i=e[i].nxt) {
int v=e[i].v;
if(dep[v]==dep[cur]+1&&e[i].w) {
flow=dfs(v,min(dis,e[i].w));
dis-=flow,sum+=flow;
e[i].w-=flow,e[i^1].w+=flow;
if(!dis) break;
}
}
return sum;
}
void solve() {
ll ans=0;
while(bfs()) {
for(int i=1;i<=n;i++) head[i]=front[i];
ans+=dfs(S);
}
vector<int> g[N];
if(ans<sum) puts("0");
else {
puts("1");
n=n-m-2;
for(int i=1;i<=m;i++) {
for(int j=front[i];j!=-1;j=e[j].nxt) {
if(e[j].v==S) continue;
else if(e[j].w==0) g[i].push_back(e[j].v-m);
}
}
for(int i=1;i<=m;i++) {
for(int j=0;j<g[i].size();j++) {
printf("%d%c",g[i][j],j+1==g[i].size()?'\n':' ');
}
}
}
}
}
int r[155],c[275];
int main() {
int m,n;
scanf("%d%d",&m,&n);
for(int i=1;i<=m;i++) {
scanf("%d",&r[i]);
sum+=r[i];
}
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
int S=m+n+1;
int T=m+n+2;
Dinic::S=S;
Dinic::T=T;
Dinic::n=n+m+2;
Dinic::m=m;
Dinic::init();
for(int i=1;i<=m;i++)
Dinic::add(S,i,r[i]);
for(int i=m+1;i<=m+n;i++)
Dinic::add(i,T,c[i-m]);
for(int i=1;i<=m;i++) {
for(int j=m+1;j<=m+n;j++) {
Dinic::add(i,j,1);
}
}
Dinic::solve();
return 0;
}