洛谷P4014 分配问题【最小/大费用流】

题目描述

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

输入格式：

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

输出格式：

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

说明

1001≤n≤100

---

题目分析

首先将讲讲最小收益

超级源点向每个工人连边
容量为1，费用为0，表示限制每个工人只能有一个工作

每个工人向每个工作连边
容量为1，费用为既定收益，这样当这条边有流时，费用即为收益

每个工作向超级汇点连边
容量为1，费用为0，表示限制每个工作只能被做一次

然后跑费用流就可以了

最大收益，即最大费用流
将原图中的边的费用全部取反
然后跑费用流，最后费用取反即可

---

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

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int inf=1128481603;
int k,n;
int s,t;
struct node{int v,f,c,nxt;}E[1000010];
int head[100010],tot=1;
int cost[100010],cnt;
int dis[100010],vis[100010];
int incf[100010],pre[100010];
int fee;

void add(int u,int v,int f,int c)
{
    E[++tot].nxt=head[u];
    E[tot].f=f;
    E[tot].v=v;
    E[tot].c=c;
    head[u]=tot;
}

bool bfs()
{
    memset(dis,67,sizeof(dis)); dis[s]=0;
    queue<int> q; q.push(s); incf[s]=inf;
    
    while(!q.empty())
    {
        int u=q.front();
        q.pop(); vis[u]=0;
        for(int i=head[u];i;i=E[i].nxt)
        {
            int v=E[i].v;
            if(E[i].f&&dis[v]>dis[u]+E[i].c)
            {
                dis[v]=dis[u]+E[i].c;
                incf[v]=min(incf[u],E[i].f); pre[v]=i;
                if(!vis[v]) q.push(v),vis[v]=1;
            }
        }
    }
    return dis[t]!=inf;
}

void dfs()
{
    int u=t;
    while(u!=s)
    {
        int i=pre[u];
        E[i].f-=incf[t]; E[i^1].f+=incf[t];
        u=E[i^1].v;
    }
    fee+=dis[t]*incf[t];
}

int main()
{
    n=read();t=n<<1|1;
    for(int i=1;i<=n;i++)
    {
    	add(s,i,1,0);add(i,s,0,0);
    	add(i+n,t,1,0);add(t,i+n,0,0);
    }
    for(int i=1;i<=n;i++)
    for(int j=n+1;j<=n*2;j++)
    {
    	cost[++cnt]=read();
    	add(i,j,1,cost[cnt]);add(j,i,0,-cost[cnt]);
   	}
    
    while(bfs())dfs();
    cout<<fee<<endl;
    
    tot=1;memset(head,0,sizeof(head));fee=cnt=0;
    for(int i=1;i<=n;i++)
    {
    	add(s,i,1,0);add(i,s,0,0);
    	add(i+n,t,1,0);add(t,i+n,0,0);
    }
    for(int i=1;i<=n;i++)
    for(int j=n+1;j<=n*2;j++)
    add(i,j,1,-cost[++cnt]),add(j,i,0,cost[cnt]);
    
    while(bfs())dfs();
    cout<<-fee<<endl;
    
    return 0;	
}