cogs 740. [网络流24题] 分配问题

1. [网络流24题] 分配问题
   ★★☆ 输入文件：job.in 输出文件：job.out 简单对比
   时间限制：1 s 内存限制：128 MB
   «问题描述：

有n件工作要分配给n个人做。第i 个人做第j 件工作产生的效益为c[i][j] 。试设计一个将
n件工作分配给n个人做的分配方案，使产生的总效益最大。
«编程任务：

对于给定的n件工作和n个人，计算最优分配方案和最差分配方案。
«数据输入：

由文件job.in提供输入数据。
文件的第1 行有1 个正整数n，表示有n件工作要分配给n 个人做。
接下来的n 行中，每行有n 个整数c[i][j] ，1≤i≤n，1≤j≤n，
表示第i 个人做第j件工作产生的效益为c[i][j] 。
«结果输出:

程序运行结束时，将计算出的最小总效益和最大总效益输出到文件job.out中。
输入文件示例 输出文件示例
job.in
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
job.out
5
14
数据范围

N<=100

---

【分析】
额…最小费用最大流&最大费用最大流的板子…写的又迷又长

---

【代码】

//最小费用最大流&最大费用最大流 
#include<iostream>
#include<climits>
#include<cstdio>
#include<queue>
#include<cstring>
#define inf 1e9+7
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=10005;
queue <int> q;
bool vis[mxn]; 
int n,m,s,t,ans,cnt;
int head[mxn],dis[mxn],pre[mxn],c[205][205];
struct node {int from,to,d,next,flow;} f[mxn<<2]; 
inline void add(int u,int v,int flow,int d) 
{
    f[++cnt].to=v,f[cnt].from=u,f[cnt].next=head[u],f[cnt].flow=flow,f[cnt].d=d,head[u]=cnt;
    f[++cnt].to=u,f[cnt].from=v,f[cnt].next=head[v],f[cnt].flow=0,f[cnt].d=-d,head[v]=cnt;
}
inline bool fsp()
{
    int i,j,d,u,v,flow,ttt;
    memset(dis,-0x3f,sizeof dis);ttt=dis[0];
    memset(vis,0,sizeof vis);
    dis[s]=0;
    q.push(s);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        vis[u]=0;
        for(i=head[u];i;i=f[i].next)
        {
            v=f[i].to,flow=f[i].flow,d=f[i].d;
            if(dis[v]<dis[u]+d && flow>0)
            {
                dis[v]=dis[u]+d;
                pre[v]=i;  //记录前驱边 
                if(!vis[v])
                  q.push(v),vis[v]=1;
            }
        }
    }
    return dis[t]>ttt;
}
inline bool spfa()
{
    int i,j,d,u,v,flow,ttt;
    memset(dis,0x3f,sizeof dis);ttt=dis[0];
    memset(vis,0,sizeof vis);
    dis[s]=0;
    q.push(s);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        vis[u]=0;
        for(i=head[u];i;i=f[i].next)
        {
            v=f[i].to,flow=f[i].flow,d=f[i].d;
            if(dis[v]>dis[u]+d && flow>0)
            {
                dis[v]=dis[u]+d;
                pre[v]=i;  //记录前驱边 
                if(!vis[v])
                  q.push(v),vis[v]=1;
            }
        }
    }
    return dis[t]<ttt;
}
inline void maxflow()
{
    int i,j,u,v,d,flow,tmp=inf;
    for(i=pre[t];i;i=pre[f[i].from])
      tmp=min(tmp,f[i].flow);
    ans+=dis[t]*tmp;
    for(i=pre[t];i;i=pre[f[i].from])
    {
        f[i].flow-=tmp;
        if(i&1) f[i+1].flow+=tmp;
        else f[i-1].flow+=tmp;
    }
}
int main()
{
    freopen("job.in","r",stdin);
    freopen("job.out","w",stdout); 
    int i,j,u,v,x,d;
    scanf("%d",&n);
    s=0,t=n+n+1;
    //最小费用最大流 
    fo(u,1,n) add(s,u,1,0);
    fo(v,n+1,n+n) add(v,t,1,0);
    fo(u,1,n) fo(v,n+1,n+n)
    {
        scanf("%d",&c[u][v]);
        add(u,v,1,c[u][v]);
    }
    while(spfa()) maxflow();
    printf("%d\n",ans);
    //最大费用最大流 
    ans=cnt=0;M(head),M(dis),M(pre);
    fo(u,1,n) add(s,u,1,0);
    fo(v,n+1,n+n) add(v,t,1,0);
    fo(u,1,n) fo(v,n+1,n+n)
      add(u,v,1,c[u][v]);
    while(fsp()) maxflow();
    printf("%d\n",ans);
    return 0;
}