分配问题(费用流)

分配问题

https://www.luogu.org/problemnew/show/P4014

题目描述

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

输入输出格式

输入格式：

文件的第 1 行有 1 个正整数n，表示有 n 件工作要分配给 n 个人做。

接下来的 n 行中，每行有 n 个整数cij​​​，表示第 i个人做第 j 件工作产生的效益为 cij​。

输出格式：

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

输入输出样例

输入样例#1： 

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

输出样例#1： 

5
14

说明

1≤n≤100

一个人只能修一个工件

把效率看成费用，因为一个人只能修一个工作，所以把容量设为1

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;

const int INF=0x3f3f3f3f;
const int N=50005;
const int M=500005;
int top;
int dist[N],pre[N];
bool vis[N];
int c[N];
int maxflow;

struct Vertex{
    int first;
}V[N];
struct Edge{
    int v,next;
    int cap,flow,cost;
}E[M];

void init(){
    memset(V,-1,sizeof(V));
    top=0;
    maxflow=0;
}

void add_edge(int u,int v,int c,int cost){
    E[top].v=v;
    E[top].cap=c;
    E[top].flow=0;
    E[top].cost=cost;
    E[top].next=V[u].first;
    V[u].first=top++;
}

void add(int u,int v,int c,int cost){
    add_edge(u,v,c,cost);
    add_edge(v,u,0,-cost);
}

bool SPFA(int s,int t,int n){
    int i,u,v;
    queue<int>qu;
    memset(vis,false,sizeof(vis));
    memset(c,0,sizeof(c));
    memset(pre,-1,sizeof(pre));
    for(i=1;i<=n;i++){
        dist[i]=INF;
    }
    vis[s]=true;
    c[s]++;
    dist[s]=0;
    qu.push(s);
    while(!qu.empty()){
        u=qu.front();
        qu.pop();
        vis[u]=false;
        for(i=V[u].first;~i;i=E[i].next){
            v=E[i].v;
            if(E[i].cap>E[i].flow&&dist[v]>dist[u]+E[i].cost){
                dist[v]=dist[u]+E[i].cost;
                pre[v]=i;
                if(!vis[v]){
                    c[v]++;
                    qu.push(v);
                    vis[v]=true;
                    if(c[v]>n){
                        return false;
                    }
                }
            }
        }
    }
    if(dist[t]==INF){
        return false;
    }
    return true;
}

int MCMF(int s,int t,int n){
    int d;
    int i,mincost;
    mincost=0;
    while(SPFA(s,t,n)){
        d=INF;
        for(i=pre[t];~i;i=pre[E[i^1].v]){
            d=min(d,E[i].cap-E[i].flow);
        }
        maxflow+=d;
        for(i=pre[t];~i;i=pre[E[i^1].v]){
            E[i].flow+=d;
            E[i^1].flow-=d;
        }
        mincost+=dist[t]*d;
    }
    return mincost;
}

int aa[105][105];

int main(){
    int n,m;
    int v,u,w,c;
    int s,t;
    cin>>n;
    init();

    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>aa[i][j];
            add(i,n+j,1,aa[i][j]);
        }
    }
    s=0,t=n+n+1;
    for(int i=1;i<=n;i++){
        add(s,i,1,0);
        add(n+i,t,1,0);
    }
    int ans=MCMF(s,t,n+n+2);
    cout<<ans<<endl;
    init();
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            add(i,n+j,1,-aa[i][j]);
        }
    }
    s=0,t=n+n+1;
    for(int i=1;i<=n;i++){
        add(s,i,1,0);
        add(n+i,t,1,0);
    }
    ans=MCMF(s,t,n+n+2);
    cout<<-ans<<endl;
}