网络流24题 第五题

欢迎访问~原文出处——博客园-zhouzhendong

去博客园看该题解

---

题目传送门 - PowerOJ1740 - 有SPJ - 推荐

题目传送门 - CodeVS1905 - 无SPJ - 0% 通过率（可以用来看题目）

---

题意概括

　　有n支队伍，m个组。第i支队伍有a[i]个人，第i个组最多可以有b[i]个人。

　　现在要求任何两个同队队员不可位于同一组，求是否有方案满足。

　　输出第一行，表示是否有，如果有，是1，没有的话，输出0；

　　如果有，接下来n行，第i行a[i]个数，表示第i支队伍的每个人被安排的组号。

　　有SPJ，只要输出任意一种方案即可。

---

题解

　　其实就是一个网络流的水题。

　　前置技能 - 网络流（传送门）

　　对于n支队伍，每只队伍一个点；对于m个组（餐桌），每个组一个点。

　　另外地，建立一个源点和一个汇点。

　　连接源点和队伍点，对于队伍i，该边的容量为a[i]；

　　连接每一个组的点和汇点，对于组i，该边的容量为b[i]；

　　对于每一个队伍，向每个组连一条边，容量为1。

　　那么图就构建完了。

　　至于证明，不解释了。

　　然后跑一跑最大流，就算出了最大匹配数。

　　其实，我们可以发现，这是一个二分图多重匹配问题。

　　如果无法全部匹配，则输出0，

　　否则输出1，再考虑。

　　然后对于连接二分图左右两端的边，如果容量为1，那么这条边就是被选择的，那么该边所连接的两个节点，“队伍点”对应“组点”，然后这样就可以把所有的匹配全部还原。

　　具体操作见代码。

---

代码

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstdlib>
using namespace std;
const int N=450,M=N*N,Inf=1<<25;
struct Edge{int x,y,cap,flow,nxt,flag;};
struct Graph{
    int cnt,s,t,n,fst[N],dist[N],cur[N],num[N],p[N],head,tail,q[N];
    Edge e[M];
    void set(int S,int T,int nn){
        s=S,t=T,n=nn,cnt=1,memset(fst,0,sizeof fst);
    }
    void add(int a,int b,int c,int d){
        e[++cnt].x=a,e[cnt].y=b,e[cnt].cap=c,e[cnt].flow=0,e[cnt].flag=d,e[cnt].nxt=fst[a],fst[a]=cnt;
        e[++cnt].x=b,e[cnt].y=a,e[cnt].cap=0,e[cnt].flow=0,e[cnt].flag=0,e[cnt].nxt=fst[b],fst[b]=cnt;
    }
    void re_bfs(){
        memset(dist,-1,sizeof dist);
        memset(q,0,sizeof q);
        head=tail=dist[t]=0;
        q[++tail]=t;
        while (head<tail)
            for (int x=q[++head],i=fst[x];i;i=e[i].nxt)
                if (e[i].cap==0&&dist[e[i].y]==-1)
                    dist[q[++tail]=e[i].y]=dist[x]+1;
        for (int i=1;i<=n;i++)
            if (dist[i]==-1)
                dist[i]=n;
    }
    int Augment(int &point){
        int ex_Flow=Inf;
        for (int i=t;i!=s;i=e[p[i]].x)
            if (ex_Flow>=e[p[i]].cap-e[p[i]].flow)
                ex_Flow=e[p[i]].cap-e[p[i]].flow,point=e[p[i]].x;
        for (int i=t;i!=s;i=e[p[i]].x)
            e[p[i]].flow+=ex_Flow,e[p[i]^1].flow-=ex_Flow;
        return ex_Flow;
    }
    int SAP(){
        int x=s,y,MaxFlow=0;
        memset(num,0,sizeof num);
        for (int i=1;i<=n;i++)
            num[dist[i]]++,cur[i]=fst[i];
        while (dist[s]<=n){
            if (x==t){
                MaxFlow+=Augment(x);
                continue;
            }
            bool found=0;
            for (int i=cur[x];i!=0&&!found;i=e[i].nxt)
                if (dist[e[i].y]+1==dist[x]&&e[i].cap>e[i].flow)
                    cur[x]=p[e[i].y]=i,x=e[i].y,found=1;
            if (found)
                continue;
            int d=n+1;
            for (int i=fst[x];i;i=e[i].nxt)
                if (e[i].cap>e[i].flow)
                    d=min(d,dist[e[i].y]+1);
            if (!(--num[dist[x]]))
                return MaxFlow;
            num[dist[x]=d]++,cur[x]=fst[x];
            if (x!=s)
                x=e[p[x]].x;
        }
        return MaxFlow;
    }
}g;
int n,m,a[N],b[N],sum=0,mat[N][N];
int main(){
    scanf("%d%d",&n,&m);
    g.set(n+m+1,n+m+2,n+m+2);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]),sum+=a[i],g.add(g.s,i,a[i],0);
    for (int i=1;i<=m;i++)
        scanf("%d",&b[i]),g.add(i+n,g.t,b[i],0);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            g.add(i,j+n,1,1);
    g.re_bfs();
    int Flow=g.SAP();
    puts(Flow<sum?"0":"1");
    if (Flow<sum)
        return 0;
    memset(mat,0,sizeof mat);
    for (int i=2;i<=g.cnt;i++)
        if (g.e[i].flag==1&&g.e[i].cap==1&&g.e[i].flow==1){
            int x=g.e[i].x,y=g.e[i].y-n;
            mat[x][++mat[x][0]]=y;
        }
    for (int i=1;i<=n;i++)
        sort(mat[i]+1,mat[i]+mat[i][0]+1);
    for (int i=1;i<=n;puts(""),i++)
        for (int j=1;j<=mat[i][0];j++)
            printf("%d ",mat[i][j]);
    return 0;
}