«问题描述:
假设有来自m 个不同单位的代表参加一次国际会议。每个单位的代表数分别为
ri(i=1,2,3...m), 。会议餐厅共有n张餐桌,每张餐桌可容纳c i(i=1,2...n) 个代表就餐。
为了使代表们充分交流,希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法,
给出满足要求的代表就餐方案。
«编程任务:
对于给定的代表数和餐桌数以及餐桌容量,编程计算满足要求的代表就餐方案。
«数据输入:
由文件roundtable.in提供输入数据。文件第1行有2 个正整数m和n,m表示单位数,n表
示餐桌数,1<=m<=150, 1<=n<=270。文件第2 行有m个正整数,分别表示每个单位的代表
数。文件第3 行有n个正整数,分别表示每个餐桌的容量。
«结果输出:
程序运行结束时,将代表就餐方案输出到文件roundtable.out中。如果问题有解,在文件第
1 行输出1,否则输出0。接下来的m行给出每个单位代表的就餐桌号。如果有多个满足要
求的方案,只要输出1 个方案。
输入文件示例 输出文件示例
roundtable.in
假设有来自m 个不同单位的代表参加一次国际会议。每个单位的代表数分别为
ri(i=1,2,3...m), 。会议餐厅共有n张餐桌,每张餐桌可容纳c i(i=1,2...n) 个代表就餐。
为了使代表们充分交流,希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法,
给出满足要求的代表就餐方案。
«编程任务:
对于给定的代表数和餐桌数以及餐桌容量,编程计算满足要求的代表就餐方案。
«数据输入:
由文件roundtable.in提供输入数据。文件第1行有2 个正整数m和n,m表示单位数,n表
示餐桌数,1<=m<=150, 1<=n<=270。文件第2 行有m个正整数,分别表示每个单位的代表
数。文件第3 行有n个正整数,分别表示每个餐桌的容量。
«结果输出:
程序运行结束时,将代表就餐方案输出到文件roundtable.out中。如果问题有解,在文件第
1 行输出1,否则输出0。接下来的m行给出每个单位代表的就餐桌号。如果有多个满足要
求的方案,只要输出1 个方案。
输入文件示例 输出文件示例
roundtable.in
4 5 4 5 3 5 3 5 2 6 4roundtable.out
1 1 2 4 5 1 2 3 4 5 2 4 5 1 2 3 4 5
题解:
建一个超级源点和一个超级汇点 源点与单位连边 容量为人数 单位与每一张桌子连边 容量为1 桌子与汇点连边容量为
桌子能坐的人数 跑最大流 如果能坐下 那么最大流就是最终能坐下的人数
输出方案:看哪一张桌子与单位之间的边流满了 就代表选了
代码:
#include <iostream> #include <cstdio> #include <vector> #include <queue> #include <cstring> using namespace std; const int maxn=500+10; const int INF=0x7fffffff; vector<int>A[maxn]; vector<int>to,cap; queue<int>q; vector<int>ans[maxn]; int dis[maxn]; int S,T; int n,m; int v[maxn],d[maxn]; int edge=0; inline bool bfs(){ memset(dis,0,sizeof(dis)); dis[S]=1; q.push(S); while(!q.empty()){ int x=q.front(); q.pop(); for(int i=0;i<A[x].size();i++){ int u=A[x][i]; if(!dis[to[u]]&&cap[u]){ dis[to[u]]=dis[x]+1; q.push(to[u]); } } } return dis[T]; } inline int dfs(int x,int mx){ if(x==T) return mx; int flow=0; for(int i=0;i<A[x].size();i++){ int u=A[x][i]; int d=0; if(cap[u]&&dis[to[u]]==dis[x]+1&&(d=dfs(to[u],min(mx,cap[u])))){ cap[u]-=d; cap[u^1]+=d; flow+=d; if(flow==mx) return flow; } } if(!flow) dis[x]=0; return flow; } inline void addedge(int f,int t,int c){ to.push_back(t); cap.push_back(c); A[f].push_back(edge++); to.push_back(f); cap.push_back(0); A[t].push_back(edge++); } int main(){ freopen("roundtable.in","r",stdin); freopen("roundtable.out","w",stdout); scanf("%d %d",&n,&m); S=0,T=n+m+1; int sum=0; for(int i=1;i<=n;i++){ scanf("%d",&v[i]); addedge(S,i,v[i]); sum+=v[i]; } for(int i=1;i<=m;i++){ scanf("%d",&d[i]); addedge(i+n,T,d[i]); } for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++) addedge(i,j+n,1); } int Ans=0; while(bfs()) Ans+=dfs(S,INF); if(Ans!=sum){ puts("0"); return 0; } puts("1"); for(int i=1;i<=n;i++){ for(int j=0;j<A[i].size();j++){ int u=A[i][j]; if(to[u]>n&&to[u]<=n+m){ if(cap[u]==0) ans[i].push_back(to[u]-n); } } } for(int i=1;i<=n;i++){ for(int j=0;j<ans[i].size();j++) printf("%d ",ans[i][j]); puts(""); } return 0; }