【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[];
int head[],cur[],dep[],n,m,s,t,u,num=-,x,y,l,tot,sum,d;
int nm,a[][];
inline int read()
{
    int x=,c=;
    char ch=' ';
    while((ch>''||ch<'')&&ch!='-')ch=getchar();
    while(ch=='-')c*=-,ch=getchar();
    while(ch<=''&&ch>='')x=x*+ch-'',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,);
}
inline bool bfs()
{
    memset(dep,,sizeof(dep));
    queue<int> q;
    while(!q.empty())
    q.pop();
    dep[s]=;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(re int i=head[u];i!=-;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(edge[i].dis>&&dep[v]==)
            {
                dep[v]=dep[u]+;
                if(v==t)
                return ;
                q.push(v);
            }
        }
    }
    return ;
}
inline int dfs(int u,int dis)
{
    if(u==t)
    return dis;
    int diss=;
    for(re int& i=cur[u];i!=-;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(edge[i].dis!=&&dep[v]==dep[u]+)
        {
            int check=dfs(v,min(dis,edge[i].dis));
            if(check>)
            {
                dis-=check;
                diss+=check;
                edge[i].dis-=check;
                edge[i^].dis+=check;
                if(dis==) break;
            }
        }
    }
    return diss;
}
inline int dinic()
{
    int ans=;
    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!=-;i=edge[i].nxt)
    {
        if(edge[i].dis==)
        cout<<edge[i].to-n<<" ";
    }
    cout<<endl;
}
int main()
{
    memset(head,-,sizeof(head));
    n=read();m=read();
    s=;t=n+m+;
    for(re int i=;i<=n;i++)
    {
        l=read();
        add(s,i,l);
        tot+=l;
    }
    for(re int i=;i<=m;i++)
    {
        l=read();
        add(i+n,t,l);
    }
    for(re int i=;i<=n;i++)
     for(re int j=;j<=m;j++)
      add(i,j+n,);
    sum=dinic();
    if(tot==sum)
    {
        cout<<""<<endl;
        for(re int i=;i<=n;i++)
        output(i);
    }else
    cout<<"";
}