[网络流24题] 圆桌问题

题目链接：戳我

好像也没有什么。对于这种匹配类问题一般不都是最大流做的嘛。。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define MAXN 100010
#define S 0
#define T n+m+1
using namespace std;
int n,m,all,t=1,cnt,all1,all2;
int head[MAXN],dis[MAXN],cur[MAXN],sum1[MAXN],sum2[MAXN];
struct Edge{int nxt,to,dis;}edge[MAXN<<1];
inline void add(int from,int to,int dis)
{
    edge[++t].nxt=head[from],edge[t].to=to,edge[t].dis=dis,head[from]=t;
    edge[++t].nxt=head[to],edge[t].to=from,edge[t].dis=0,head[to]=t;
}
inline bool bfs()
{
    queue<int>q;
    memset(dis,0x3f,sizeof(dis));
    memcpy(cur,head,sizeof(head));
    q.push(S);dis[S]=0;
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=head[u];i;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(edge[i].dis&&dis[v]==0x3f3f3f3f)
                dis[v]=dis[u]+1,q.push(v);
        }
    }
    if(dis[T]==0x3f3f3f3f) return false;
    return true;
}
inline int dfs(int x,int f)
{
    if(!f||x==T) return f;
    int used=0,w;
    for(int i=cur[x];i;i=edge[i].nxt)
    {
        cur[x]=i;
        int v=edge[i].to;
        if(dis[v]==dis[x]+1&&(w=dfs(v,min(f,edge[i].dis))))
        {
            edge[i].dis-=w,edge[i^1].dis+=w;
            f-=w,used+=w;
            if(!f) break;
        }
    }
    return used;
}
inline int dinic()
{
    int cur_ans=0;
    while(bfs())
        cur_ans+=dfs(0,(int)1e9);
    return cur_ans;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&sum1[i]),all+=sum1[i];
    for(int i=1;i<=m;i++)
        scanf("%d",&sum2[i]);
    for(int i=1;i<=n;i++)
        add(S,i,sum1[i]);
    for(int i=1;i<=m;i++)
        add(i+n,T,sum2[i]);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            add(i,j+n,1);
    if(dinic()>=all) printf("1
");
    else {printf("0
"); return 0;}
    for(int i=1;i<=n;i++)
    {
        for(int j=head[i];j;j=edge[j].nxt)
        {
            if(edge[j^1].dis!=0)
                printf("%d ",edge[j].to-n);
        }
        printf("
");
    }
}