【P3254】圆桌问题(最大流，洛谷)

看到题目，产生第一反应，是否可以匹配的是这么多。那么连边跑一遍最大流就行了。

从源点向每个单位连一条长度为l的边，然后所有单位和餐桌分别连边，流量为1，所有餐桌向汇点连边，流量为餐桌容量。
然后跑一遍最大流即可。

输出方案的时候直接暴力枚举一个单位的所有出边即可，如果不为0，那么说明使用过，即可输出。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
#define inf 50000000
#define re register
#define id n*(i-1)+j
using namespace std;
struct po
{
    int from,to,dis,nxt;
}edge[500001];
int head[500001],cur[500001],dep[50002],n,m,s,t,u,num=-1,x,y,l,tot,sum,d;
int nm,a[210][210];
inline int read()
{
    int x=0,c=1;
    char ch=' ';
    while((ch>'9'||ch<'0')&&ch!='-')ch=getchar();
    while(ch=='-')c*=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=getchar();
    return x*c;
}
inline void add_edge(int from,int to,int dis)
{
    edge[++num].nxt=head[from];
    edge[num].from=from;
    edge[num].to=to;
    edge[num].dis=dis;
    head[from]=num;
}
inline void add(int from,int to,int dis)
{
    add_edge(from,to,dis);
    add_edge(to,from,0);
}
inline bool bfs()
{
    memset(dep,0,sizeof(dep));
    queue<int> q;
    while(!q.empty())
    q.pop();
    dep[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(re int i=head[u];i!=-1;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(edge[i].dis>0&&dep[v]==0)
            {
                dep[v]=dep[u]+1;
                if(v==t)
                return 1;
                q.push(v);
            }
        }
    }
    return 0;
}
inline int dfs(int u,int dis)
{
    if(u==t)
    return dis;
    int diss=0;
    for(re int& i=cur[u];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(edge[i].dis!=0&&dep[v]==dep[u]+1)
        {
            int check=dfs(v,min(dis,edge[i].dis));
            if(check>0)
            {
                dis-=check;
                diss+=check;
                edge[i].dis-=check;
                edge[i^1].dis+=check;
                if(dis==0) break;
            }
        }
    }
    return diss;
}
inline int dinic()
{
    int ans=0;
    while(bfs())
    {
        for(re int i=s;i<=t;i++)
        cur[i]=head[i];
        while(int d=dfs(s,inf))
        ans+=d;
    }
    return ans;
}
inline void output(int x)
{
    for(re int i=head[x];i!=-1;i=edge[i].nxt)
    {
        if(edge[i].dis==0)
        cout<<edge[i].to-n<<" ";
    }
    cout<<endl;
}
int main()
{
    memset(head,-1,sizeof(head));
    n=read();m=read();
    s=0;t=n+m+1;
    for(re int i=1;i<=n;i++)
    {
        l=read();
        add(s,i,l);
        tot+=l;
    }
    for(re int i=1;i<=m;i++)
    {
        l=read();
        add(i+n,t,l);
    }
    for(re int i=1;i<=n;i++)
     for(re int j=1;j<=m;j++)
      add(i,j+n,1);
    sum=dinic();
    if(tot==sum)
    {
        cout<<"1"<<endl;
        for(re int i=1;i<=n;i++)
        output(i);
    }else
    cout<<"0";
}