圆桌问题 2011-12-29

算法实现题8-5 圆桌问题（习题 8-15）
´问题描述：
假设有来自 n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为
n i ri
, , 2 , 1 ,  = 。会议餐厅共有 m张餐桌，每张餐桌可容纳 ) , , 2 , 1 ( m i ci
 = 个代表就餐。
为了使代表们充分交流， 希望从同一个单位来的代表不在同一个餐桌就餐。 试设计一个算法，
给出满足要求的代表就餐方案。
´编程任务：
对于给定的代表数和餐桌数以及餐桌容量，编程计算满足要求的代表就餐方案。
´数据输入：
由文件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 

 

————————————————————

Program Stone;
var flow,n,m,s,t,tot:longint;
    map:array[0..500,0..500]of longint;
    dis,cur,vh:array[0..1000]of longint;
 procedure init;
 var i,j,k:longint;
  begin
    readln(m,n);
    s:=0;t:=m+n+1;
    for i:=1 to m do
     begin
       read(j);
       map[s,i]:=j;
       inc(tot,j);
     end;
    for i:=1 to n do
     begin
       read(j);
       map[i+m,t]:=j;
     end;
    for i:=1 to m do
     for j:=1 to n do
       map[i,j+m]:=1;
  end;
 function min(x,y:longint):longint;
  begin
   if x<y then min:=x else min:=y;
  end;
 function aug(x,nf:longint):longint;
 var i,j,d,l,minh,ins:longint;
  begin
    if x=t then exit(nf);
    l:=nf;
    for i:=cur[x] to t do
     if (map[x,i]>0)and(dis[i]+1=dis[x]) then
     begin
         cur[x]:=i;
         d:=aug(i,min(l,map[x,i]));
         dec(map[x,i],d);
         inc(map[i,x],d);
         dec(l,d);
         if (dis[s]=t+1)or(l=0) then exit(nf-l);
     end;
    if l=nf then
     begin
       minh:=t+1;
       for i:=s to t do
         if (map[x,i]>0)and(minh>dis[i]) then begin minh:=dis[i];ins:=i;end;
       cur[x]:=ins;
       dec(vh[dis[x]]);
       if vh[dis[x]]=0 then dis[s]:=t+1;
       dis[x]:=minh+1;
       inc(vh[dis[x]]);
     end;
    aug:=nf-l;
  end;
 procedure print;
 var i,j:longint;
  begin
    if flow<>tot then write(0)
                 else begin
                       writeln(1);
                       for i:=1 to m do
                       begin
                        for j:=1 to n do
                         if map[i,j+m]=0 then write(j,' ');
                        writeln;
                       end;
                      end;
  end;
Begin
 assign(input,'prog85.in');assign(output,'prog85.out');
 reset(input);rewrite(output);
  init;
  vh[0]:=t+2;
  while dis[s]<t+1 do inc(flow,aug(s,maxint));
  print;
 close(input);close(output);
end.