圆桌问题

问题描述

    假设有来自 n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为ri,i=1,2,···,n。会议餐厅共有m张餐桌，每张餐桌可容纳ci
(i=1,2,···,m)个代表就餐。为了使代表们充分交流， 希望从同一个单位来的代表不在同一个餐桌就餐。 试设计一个算法，给出满足要求的代表就
餐方案。

注：应当是m个单位，n张餐桌，原题有误。

编程任务

    对于给定的代表数和餐桌数以及餐桌容量，编程计算满足要求的代表就餐方案。

数据输入

    由文件input.txt提供输入数据。文件第1行有2个正整数m和n，m表示单位数，n表示餐桌数1<=m<=150, 1<=n<=270。文件第2行有m个正整数，分别
表示每个单位的代表数。文件第3行有n个正整数，分别表示每个餐桌的容量。

结果输出

    程序运行结束时，将代表就餐方案输出到文件 output.txt 中。如果问题有解，在文件第1行输出1，否则输出0。接下来的m行给出每个单位代表的
就餐桌号。如果有多个满足要求的方案，只要输出1个方案。

输入文件示例

input.txt 
4 5 
4 5 3 5 
3 5 2 6 4

输出文件实例

output.txt 
1 
1 2 4 5 
1 2 3 4 5
2 4 5 
1 2 3 4 5

题解

这题真是水的一比。（我一开始没有找到合适的OJ，所以没做，然而COGS上的题目名称是圆桌聚餐，好坑啊）
对于每个单位，我们都从源点引一条边，容量为该单位的人数。然后从每个单位，向每一个餐桌引一条容量为1的边。对于每个餐桌，我们都进行拆点（其实不拆点，用二分图做也可以，但我当时没有想到这一点），连接对应两点的边的容量为餐桌可容纳的人数。跑一边dinic（不需要返回结果），从源点进行查看，是否存在有连接到一个单位的边仍未满流，如果存在就输出0，结束，否则输出1（我忘记输出1还能在COGS上得60分，神奇）。然后枚举每一个单位，看其连向每个餐桌的边的流量是否耗尽，是的话就统计到答案中（一定要排序后在输出）。

CODE：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=1e9;
struct edge
{
    int next,to,remain;
}a[20000];
int head[10000];
int deep[10000];
int ans[10000];
int n,m,x,num=1,S,T,tot;
bool b=1;
queue<int>q;
inline int min(int a,int b){return a<b?a:b;}
inline void add(int x,int y,int cap)
{
    a[++num].next=head[x],a[num].to=y,a[num].remain=cap,head[x]=num;
    a[++num].next=head[y],a[num].to=x,head[y]=num;
}
inline bool bfs()
{
    memset(deep,0x3f,sizeof(deep));
    q.push(S);deep[S]=0;
    while(!q.empty())
    {
        int tmp=q.front();q.pop();
        for(int i=head[tmp];i;i=a[i].next)
          if(deep[a[i].to]>INF&&a[i].remain) q.push(a[i].to),deep[a[i].to]=deep[tmp]+1;
    }
    return deep[T]<INF;
}
int dfs(int now,int limit)
{
    if(now==T||!limit) return limit;
    int flow=0,f;
    for(int i=head[now];i;i=a[i].next)
      if(deep[a[i].to]==deep[now]+1&&a[i].remain&&(f=dfs(a[i].to,min(limit,a[i].remain))))
      {
        flow+=f,limit-=f,a[i].remain-=f,a[i^1].remain+=f;
        if(!limit) return flow;
      }
    deep[now]=-1;return flow;
}
inline void dinic() {while(bfs()) dfs(S,INF);}
int main()
{
    scanf("%d%d",&m,&n);
    //1~m:单位 m+1~2*m+n:餐桌 m+n+1~m+2*n:餐桌拆点 
    S=m+(n<<1)+1,T=S+1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&x),add(S,i,x);
        for(int j=1;j<=n;j++)
          add(i,m+j,1);
    }
    for(int i=1;i<=n;i++)
      scanf("%d",&x),add(m+i,m+n+i,x);
    for(int i=1;i<=n;i++)
      add(m+n+i,T,INF);
    dinic();
    for(int i=head[S];i;i=a[i].next)
      if(a[i].remain) {b=0;break;}
    if(!b) {printf("0");return 0;}
    else printf("1\n");
    for(int i=1;i<=m;i++)
    {
        memset(ans,0,sizeof(ans));
        tot=0;
        for(int j=head[i];j;j=a[j].next)
          if(!a[j].remain) ans[++tot]=a[j].to-m;
        sort(ans+1,ans+tot+1);
        for(int i=1;i<=tot;i++)
          printf("%d ",ans[i]);
        printf("\n");
    }
    return 0;
}