UESTC 1962 天才钱vs学霸周2 最大流

题目：http://www.qscoj.cn/#/problem/show/1962

 

建立一个源点s和汇点t

每个点预置1后流量最小值就变成了0

将a[i]连接到s,b[i]连接到t

再将每行的点连接到每列，流量为19

再跑一次dinic

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
using namespace std;
const int N=1e5+5;
const int inf=0x3f3f3f3f;
int c[105][105];
int a[N],b[N];
int n,m;
struct Dinic
{
    int s,t,tot;
    int head[N],nt[N],to[N],val[N],dep[N],cur[N];
    void init(int ss,int tt)
    {
        s=ss;
        t=tt;
        tot=-1;
        memset(head,-1,sizeof(head));
        memset(nt,-1,sizeof(nt));
    }
    void addedge(int u,int v,int w)
    {
        nt[++tot]=head[u];
        to[tot]=v;
        val[tot]=w;
        head[u]=tot;
        nt[++tot]=head[v];
        to[tot]=u;
        val[tot]=0;
        head[v]=tot;
    }
    bool bfs()
    {
        queue<int> que;
        while(!que.empty())
            que.pop();
        memset(dep,0,sizeof(dep));
        dep[s]=1;
        que.push(s);
        while(!que.empty())
        {
            int u=que.front();
            que.pop();
            for(int i=head[u];i!=-1;i=nt[i])
                if (val[i]>0&&dep[to[i]]==0)
                {
                    dep[to[i]]=dep[u]+1;
                    que.push(to[i]);
                }
        }
        if (dep[t]==0) return 0;
        return 1;
    }
    int dfs(int u,int dis)
    {
        if (u==t) return dis;
        for(int &i=cur[u];i!=-1;i=nt[i])
            if (dep[to[i]]==dep[u]+1&&val[i]>0)
            {
                int di=dfs(to[i],min(dis,val[i]));
                if (di>0)
                {
                    val[i]-=di;
                    val[i^1]+=di;
                    return di;
                }
            }
        return 0;
    }
    int dinic()
    {
        int ans=0;
        while(bfs())
        {
            for(int i=s;i<=t;i++)
                cur[i]=head[i];
            while(int d=dfs(s,inf))
                ans+=d;
        }
        int flag=1;
        int s=0;
        for(int i=1;i<=n;i++) s+=a[i];
        if (ans!=s) flag=0;
        if (flag)
        {
            printf("Yes\n");
            for(int i=1;i<=n;i++)
                for(int j=head[i];j!=-1;j=nt[j])
                    c[i][to[j]-n]=20-val[j];
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<m;j++)
                    printf("%d ",c[i][j]);
                printf("%d\n",c[i][m]);
            }
        }
        else printf("No\n");
        return 0;
    }
};

Dinic dinic;
int main()
{
    scanf("%d%d",&n,&m);
    int t1=0,t2=0;
    for(int i=1;i<=n;i++) {scanf("%d",&a[i]);a[i]-=m;t1+=a[i];}
    for(int i=1;i<=m;i++) {scanf("%d",&b[i]);b[i]-=n;t2+=b[i];}
    if (t1!=t2)
    {
        printf("No\n");
        return 0;
    }
    dinic.init(0,n+m+1);
    for(int i=1;i<=n;i++)
        dinic.addedge(0,i,a[i]);
    for(int i=1;i<=m;i++)
        dinic.addedge(i+n,n+m+1,b[i]);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        dinic.addedge(i,n+j,19);
    dinic.dinic();
    return 0;
}