首页 诗词 字典 板报 句子 名言 友答 励志 学校 网站地图
当前位置: 首页 > 教程频道 > 开发语言 > 编程 >

POJ 3422 最小开销最大流 zkw或者普通版本

2012-09-17 
POJ 3422 最小费用最大流 zkw或者普通版本建图的话每个点拆成两个点u, u,连一条容量为1费用为金币数的边,

POJ 3422 最小费用最大流 zkw或者普通版本

建图的话

每个点拆成两个点u, u',连一条容量为1费用为金币数的边,再连一条容量为k,费用为0的边

然后每个点和他右边或者下边的点连边 i'->j这样连

然后源点连1点,右下角那个点去连汇点,容量都为k,费用为0

普通写法

#include <iostream>#include <algorithm>#include <cstring>#include <string>#include <cstdio>#include <cmath>#include <queue>#include <deque>#include <map>#include <set>#define eps 1e-5#define MAXN 5555#define MAXM 55555#define INF 100000007using namespace std;struct EDGE{    int cost, cap, v;    int next, re;}edge[MAXM];int head[MAXN], e;int vis[MAXN], d[MAXN];int ans, cost, src, des, n;void init(){    memset(head, -1, sizeof(head));    e = 0;    ans = cost = 0;}void add(int u, int v, int cap, int cost){    edge[e].v = v;    edge[e].cap = cap;    edge[e].cost = cost;    edge[e].re = e + 1;    edge[e].next = head[u];    head[u] = e++;    edge[e].v = u;    edge[e].cap = 0;    edge[e].cost = -cost;    edge[e].re = e - 1;    edge[e].next = head[v];    head[v] = e++;}int aug(int u, int f){    if(u == des)    {        ans += cost * f;        return f;    }    vis[u] = 1;    int tmp = f;    for(int i = head[u]; i != -1; i = edge[i].next)        if(edge[i].cap && !edge[i].cost && !vis[edge[i].v])        {            int delta = aug(edge[i].v, tmp < edge[i].cap ? tmp : edge[i].cap);            edge[i].cap -= delta;            edge[edge[i].re].cap += delta;            tmp -= delta;            if(!tmp) return f;        }    return f - tmp;}bool modlabel(){    for(int i = 0; i <= n; i++) d[i] = INF;    d[des] = 0;    deque<int>Q;    Q.push_back(des);    while(!Q.empty())    {        int u = Q.front(), tmp;        Q.pop_front();        for(int i = head[u]; i != -1; i = edge[i].next)            if(edge[edge[i].re].cap && (tmp = d[u] - edge[i].cost) < d[edge[i].v])                (d[edge[i].v] = tmp) <= d[Q.empty() ? src : Q.front()] ? Q.push_front(edge[i].v) : Q.push_back(edge[i].v);    }    for(int u = 1; u <= n; u++)        for(int i = head[u]; i != -1; i = edge[i].next)            edge[i].cost += d[edge[i].v] - d[u];    cost += d[src];    return d[src] < INF;}void costflow(){    while(modlabel())    {        do        {            memset(vis, 0, sizeof(vis));        }while(aug(src, INF));    }}int nt, k;int main(){    int w;    while(scanf("%d%d", &nt, &k) != EOF)    {        init();        for(int i = 1; i <= nt; i++)            for(int j = 1; j <= nt; j++)            {                scanf("%d", &w);                int id = (i - 1) * nt + j;                add(id, id + nt * nt, 1, -w);                add(id, id + nt * nt, k, 0);                if(i < nt) add(id + nt * nt, id + nt, k, 0);                if(j < nt) add(id + nt * nt, id + 1, k, 0);            }        src = nt * nt * 2 + 1;        des = nt * nt * 2 + 2;        n = des;        add(src, 1, k, 0);        add(nt * nt * 2, des, k, 0);        costflow();        printf("%d\n", -ans);    }    return 0;}


热点排行