loj6012「网络流 24 题」分配问题

(http://www.elijahqi.win/2017/12/15/loj6012%E3%80%8C%E7%BD%91%E7%BB%9C%E6%B5%81-24-%E9%A2%98%E3%80%8D%E5%88%86%E9%85%8D%E9%97%AE%E9%A2%98/)
题目描述

有 n n n 件工作要分配给 n n n 个人做。第 i i i 个人做第 j j j 件工作产生的效益为 cij c_{ij} c​ij​​。试设计一个将 n n n 件工作分配给 n n n 个人做的分配方案，使产生的总效益最大。
输入格式

文件的第 1 1 1 行有 1 1 1 个正整数 n n n，表示有 n n n 件工作要分配给 n n n 个人做。接下来的 n n n 行中，每行有 n n n 个整数 cij c_{ij} c​ij​​，表示第 i i i 个人做第 j j j 件工作产生的效益为 cij c_{ij} c​ij​​。
输出格式

两行分别输出最小总效益和最大总效益。
样例
样例输入

5
2 2 2 1 2
2 3 1 2 4
2 0 1 1 1
2 3 4 3 3
3 2 1 2 1

样例输出

5
14

数据范围与提示

1≤n≤100 1 \leq n \leq 100 1≤n≤100

和上一题思路基本相同

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define inf 0x3f3f3f3f
#define N 110
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9') ch=gc();
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=gc();}
    return x;
}
struct node{
    int y,z,next,c;
}data[N*N];
int h[N],f[N],n,T,mp[N][N],num=1,pre[N],path[N];bool flag[N];
inline void insert1(int x,int y,int z,int c){
    data[++num].y=y;data[num].z=z;data[num].c=c;data[num].next=h[x];h[x]=num;
    data[++num].y=x;data[num].z=0;data[num].c=-c;data[num].next=h[y];h[y]=num;
}
inline bool spfa(){
    queue<int>q;memset(f,0x3f,sizeof(f));memset(flag,0,sizeof(flag));flag[0]=1;f[0]=0;q.push(0);memset(pre,-1,sizeof(pre));
    while(!q.empty()){
        int x=q.front();q.pop();flag[x]=0;
        for (int i=h[x];i;i=data[i].next){
            int y=data[i].y,z=data[i].z,c=data[i].c;
            if (f[x]+c<f[y]&&z){
                f[y]=f[x]+c;pre[y]=x;path[y]=i;
                if (!flag[y]) flag[y]=1,q.push(y);
            }
        }
    }if(pre[T]==-1) return 0;else return 1;
}
inline bool spfa1(){
    queue<int>q;memset(f,128,sizeof(f));memset(flag,0,sizeof(flag));flag[0]=1;f[0]=0;q.push(0);memset(pre,-1,sizeof(pre));
    while(!q.empty()){
        int x=q.front();q.pop();flag[x]=0;
        for (int i=h[x];i;i=data[i].next){
            int y=data[i].y,z=data[i].z,c=data[i].c;
            if (f[x]+c>f[y]&&z){
                f[y]=f[x]+c;pre[y]=x;path[y]=i;
                if (!flag[y]) flag[y]=1,q.push(y);
            }
        }
    }if(pre[T]==-1) return 0;else return 1;
}
int main(){
    freopen("loj.in","r",stdin);
    n=read();T=n*2+1;
    for (int i=1;i<=n;++i) insert1(0,i,1,0),insert1(i+n,T,1,0);
    for (int i=1;i<=n;++i)
        for (int j=1;j<=n;++j) mp[i][j]=read(),insert1(i,j+n,1,mp[i][j]);
    int ans=0;
    while(spfa()){
        int minn=inf,now=T;
        while(now){minn=min(minn,data[path[now]].z),now=pre[now];}now=T;
        while(now){ans+=data[path[now]].c*minn;data[path[now]].z-=minn;data[path[now]^1].z+=minn;now=pre[now];}
    }printf("%d\n",ans);ans=0;memset(h,0,sizeof(h));
    for (int i=1;i<=n;++i) insert1(0,i,1,0),insert1(i+n,T,1,0);
    for (int i=1;i<=n;++i)
        for (int j=1;j<=n;++j) insert1(i,j+n,1,mp[i][j]);
    while(spfa1()){
        int minn=inf,now=T;
        while(now){minn=min(minn,data[path[now]].z),now=pre[now];}now=T;
        while(now){ans+=data[path[now]].c*minn;data[path[now]].z-=minn;data[path[now]^1].z+=minn;now=pre[now];}
    }printf("%d",ans);
    return 0;
}