#6004. 「网络流 24 题」圆桌聚餐
内存限制:256 MiB
时间限制:5000 ms
标准输入输出
题目类型:传统
评测方式:Special Judge
上传者: 匿名
题目描述
输入格式
输出格式
样例
数据范围与提示
如果代表团到那个桌子的流量为0 即选了那个桌子
#include <iostream>
#include <cstdio>
#include <cstring>
#include <climits>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 2100;
const int inf = 99999999;
int dis[N], lter[N];
struct node
{
int to, cap, rev, cap2, id;
};
vector<node>p[N];
void add(int u,int v,int w)
{
node tmp;
tmp.to=v, tmp.cap=w,tmp.rev=p[v].size();
p[u].push_back(tmp);
tmp.to=u, tmp.cap=0,tmp.rev=p[u].size()-1;
p[v].push_back(tmp);
return ;
}
int bfs(int s,int t)
{
memset(dis,-1,sizeof(dis));
queue<int>q;
dis[s]=0;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=0; i<p[u].size(); i++)
{
int v=p[u][i].to;
if(dis[v]==-1&&p[u][i].cap>0)
{
dis[v]=dis[u]+1;
q.push(v);
}
}
}
return dis[t]==-1;
}
int dfs(int s,int t,int flow)
{
if(s==t) return flow;
for(int &i=lter[s]; i<p[s].size(); i++)
{
int v=p[s][i].to;
if(dis[v]==dis[s]+1&&p[s][i].cap>0)
{
int f=dfs(v,t,min(flow,p[s][i].cap));
if(f>0)
{
p[s][i].cap-=f, p[v][p[s][i].rev].cap+=f;
return f;
}
}
}
return 0;
}
int dinic(int s,int e)
{
int flow=0, f;
while(1)
{
while(bfs(s,e)) return flow;
memset(lter,0,sizeof(lter));
while(f=dfs(s,e,inf),f>0)
{
flow+=f;
}
}
return flow;
}
int a[N], b[N];
int main()
{
int n, m;
scanf("%d %d", &m, &n);
int s=0, t=m+n+1, sum=0;
for(int i=1;i<=m;i++)
{
scanf("%d",&a[i]);
add(s,i,a[i]);
sum+=a[i];
}
for(int i=1;i<=n;i++)
{
scanf("%d",&b[i]);
add(m+i,t,b[i]);
}
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++) add(i,j+m,1);
int ans=dinic(s,t);
if(ans<sum)puts("0");
else
{
puts("1");
for(int i=1;i<=m;i++)
{
for(int j=0;j<p[i].size();j++)
if(p[i][j].cap==0) printf("%d ",p[i][j].to-m);
puts("");
}
}
return 0;
}