网络流24题5 圆桌问题

分析

先放一个可以测试的网站——神奇的题库
然后看题：
建图的过程
1. 添加源点s和汇点t
2. 在s与每一个单位i之间连接一条容量为ri的有向边
3. 在每一个单位i与每一个餐桌j之间连接一条容量为1的有向边(因为每张桌子上不能有同一单位的
4. 在每一个餐桌j与t连接一条容量为ci的有向边

然后求最大流，如果从s出发的每一条边全部满载的话问题就有解（全部满载表示每一个代表都有一个座位），否则问题就无解。至于在有解时打印方案的话可以直接遍历每一个单位i，有流量的边连向的餐桌就表示有有一个代表在哪个餐桌就餐。

ps：要用当前弧优化，不然会超时。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>

using namespace std;

struct arr{
    int x,y,w;
    int op,next;
}edge[300000];
int ls[501];
int cur[501];
int w[150001];
int queue[501];
int dis[501];
int q[501];
int s,t;
int n,m,nn;
int min1,ans;
int sum;

void er(int ww){w[nn]=ww;}

void add(int x,int y,int w)
{
    nn++; 
    edge[nn].x=x; edge[nn].y=y; edge[nn].w=w;
    edge[nn].op=nn+1; edge[nn].next=ls[x];
    ls[x]=nn; er(w); cur[x]=ls[x];

    nn++;
    edge[nn].x=y; edge[nn].y=x; edge[nn].w=w;
    edge[nn].op=nn-1; edge[nn].next=ls[y];
    ls[y]=nn; er(0); cur[y]=ls[y];
}

bool bfs()
{
    memset(dis,-1,sizeof(dis));
    int head,tail;
    head=0; tail=1;
    queue[head]=0;
    dis[0]=0;
    do
    {
        head++;
        int i=ls[queue[head]];
        while (i!=0)
        {
            if ((w[i]>0)&&(dis[edge[i].y]==-1))
            {
                dis[edge[i].y]=dis[edge[i].x]+1;
                tail++;
                queue[tail]=edge[i].y;
                if (edge[i].y==t)  return true;
            }
            i=edge[i].next;
        }
    }while (head<tail);
    return false;
}

bool find(int x,int num)
{
    if (x!=s) min1=min(min1,w[num]);
    if (x==t) return true;
    for (int &i=cur[x];i!=0;i=edge[i].next)
    {
        int x=edge[i].x;
        int y=edge[i].y;
        if ((dis[x]+1==dis[y])&&(w[i]!=0)&&(find(y,i)))
        {
            w[i]-=min1;
            w[edge[i].op]+=min1;
            return true;
        }
    }
    return false;
}

int dinic()
{
    while (bfs())
    {
        for (int i=s;i<=t;i++)
            cur[i]=ls[i];
        min1=2000000000;
        find(s,0);
        ans-=min1;
    }
    if (ans!=0)
        printf("%d\n",0);
    else
    {
        printf("%d\n",1);
        for (int i=1;i<=n;i++)
        {
            q[0]=0;
            for (int j=ls[i];j!=0;j=edge[j].next)
                if (w[j]==0) q[++q[0]]=edge[j].y-n;
            for (int i=q[0];i>0;i--)
                printf("%d ",q[i]);
            printf("\n");
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    s=0; t=n+m+1;
    ans=0;
    for (int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        ans+=x;
        add(s,i,x);
        for (int j=n+1;j<=n+m;j++) add(i,j,1);
    }
    for (int i=n+1;i<=n+m;i++)
    {
        int x;
        scanf("%d",&x);
        add(i,t,x);
    }
    dinic();
    return 0;
}