首页 > 其他 > 详细

ACM/ICPC 之 卡卡的矩阵旅行-最小费用最大流(可做模板)(POJ3422)

时间:2016-07-29 18:31:23      阅读:180      评论:0      收藏:0      [点我收藏+]

  将每个点拆分成原点A与伪点B,A->B有两条单向路(邻接表实现时需要建立一条反向的空边,并保证环路费用和为0),一条残留容量为1,费用为本身的负值(便于计算最短路),另一条残留容量+∞,费用为0(保证可以多次通过该点,但费用只计算一次)。

  另外伪点B与原点右侧与下方的点有一条单向路(邻接表实现需要建立反向空边),残留容量为+∞,费用为0。源点0到点1有一条单向路,残留容量为K(可以通过K次),最后一个点的伪点2*n*n与汇点2*n*n+1有一条单向边,残留容量为K,两条边的费用都为0。

  构图成功后,走一次最小费用最大流即可。

 

//最小费用最大流问题-拆点
//Time:47Ms     Memory:624K
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
#define MAX 55
#define MAXN 5005
#define INF 0x3f3f3f3f
struct Edge{
    int u,v,r,w,next;
    Edge(){}
    Edge(int U,int V,int R,int W,int N):u(U),v(V),r(R),w(W),next(N){}
}e[MAXN*8];
int n,k;
int s,t;
int h[MAXN],le;
int m[MAX][MAX];
int pre[MAXN];
int cost[MAXN];
bool vis[MAXN];
void add(int u,int v,int r,int w)
{
    e[le] = Edge(u,v,r,w,h[u]);   h[u] = le++;
    e[le] = Edge(v,u,0,-w,h[v]);   h[v] = le++;
}
void build()
{
    add(s, 1, k, 0);
    add(2*n*n, t, k, 0);
    for(int i = 1; i <= n*n; i++)
    {
        add(i, n*n+i, 1, -m[(i - 1)/n + 1][(i - 1)%n + 1]);
        add(i, n*n+i, k, 0);
        if(i % n)   add(i+n*n, i+1, INF, 0);
        if(i <= n*(n-1))    add(i+n*n, i+n, INF, 0);
    }
}
bool spfa()
{
    memset(pre,-1,sizeof(pre));
    memset(vis,false,sizeof(vis));
    memset(cost,INF,sizeof(cost));
    queue<int> q;
    q.push(s);
    cost[s] = 0;    vis[s] = true;
    while(!q.empty()){
        int cur = q.front();
        q.pop();    vis[cur] = false;
        for(int i = h[cur]; i != -1; i = e[i].next)
        {
            int v = e[i].v;
            if(e[i].r && cost[v] > cost[cur] + e[i].w){
                cost[v] = cost[cur] + e[i].w;
                pre[v] = i;
                if(!vis[v]){
                    vis[v] = true;  q.push(v);
                }
            }
        }
    }
    return cost[t] == INF? false: true;
}
int maxcostflow()
{
    int maxFlow = 0;
    while(spfa()){  //增广路
        int mind = INF;
        for(int i = pre[t]; i != -1; i = pre[e[i].u])
            mind = min(mind, e[i].r);
        for(int i = pre[t]; i != -1; i = pre[e[i].u])
        {
            e[i].r -= mind;
            e[i^1].r += mind;
        }
        maxFlow += mind*cost[t];
    }
    return maxFlow;
}
int main()
{
    //freopen("in.txt", "r", stdin);
    memset(h, -1, sizeof(h));
    scanf("%d%d", &n,&k);
    s = 0; t = 2*n*n+1;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
            scanf("%d", &m[i][j]);
    build();
    printf("%d\n", -maxcostflow());
    return 0;
}

 

ACM/ICPC 之 卡卡的矩阵旅行-最小费用最大流(可做模板)(POJ3422)

原文:http://www.cnblogs.com/Inkblots/p/5719179.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!