POJ 3422 费用流

这题和HDU 3376类似。只不过HDU那题是从左上走到右下，再走回来。这题是从左上到右下走K次。只是在建图上有略微的区别。

建图：

S -> 1 ，流量K，费用0，同理n * n + n * n -> T 。

拆点i -> i' ，流量1，费用为该点的值，需注意因为该题点可以重复走，但是第二次走该点时值为0，所以需要再加一条流量k-1,费用0的边。

如果i,j可达，i' -> j ,流量k，费用0 。

Code:

///poj 3422
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 10005
#define inf 1<<28
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
using namespace std;

struct kdq
{
    int  s , e , l ,next ,c ;
}ed[Max * 100] ;
int head[Max] , num ;
void add(int s, int e ,int l ,int c)
{
    ed[num].s = s ;
    ed[num].e = e ;
    ed[num].l = l ;
    ed[num].c = c ;
    ed[num].next = head[s] ;
    head[s] = num ++ ;

    ed[num].s = e ;
    ed[num].e = s ;
    ed[num].l = 0 ;
    ed[num].c = -c ;
    ed[num].next = head[e] ;
    head[e] = num ++ ;
}
void init()
{
    mem(head,-1) ;
    num = 0 ;
}
int n , k ;
int Map[55][55] ;
int mx[2] = {0,1} ;
int my[2] = {1,0} ;
int inmap(int x ,int y)
{
    if(x >= 1 && x <= n && y >= 1 && y <= n)return 1  ;
    return 0 ;
}
int S, T ;
void build()
{
    S = 0 , T = n * n + n * n + 1 ;
    REP(i,1,n)
    {
        REP(j,1,n)
        {
            add((i - 1) * n + j ,(i - 1) * n + j + n * n ,1,Map[i][j]) ;
            add((i - 1) * n + j ,(i - 1) * n + j + n * n ,k - 1,0) ;
            REP(k,0,1)
            {
                int tx = i + mx[k] ;
                int ty = j + my[k] ;
                if(inmap(tx,ty))
                {
                    add((i - 1) * n + j + n * n ,(tx - 1) * n + ty ,inf, 0) ;
                }
            }
        }
    }
    add(S,1,k,0) ;
    add(n * n + n * n ,T,k,0) ;
}
int dis[Max] ;
bool vis[Max] ;
int qe[Max * 100] ;
int path[Max] ;
int pre[Max] ;
int spfa()
{
    REP(i,0,T)dis[i] = -1 ,vis[i] = 0 ;
    dis[S] = 0 ;
    vis[S] = 1 ;
    int h = 0 ;
    int t = 0 ;
    qe[h ++ ] = S ;
    while(h > t)
    {
        int tt = qe[t ++ ] ;
        vis[tt] = 0 ;
        for (int i = head[tt] ; ~i ; i = ed[i].next )
        {

            int e = ed[i].e ;
            int l = ed[i].l ;
            int c = ed[i].c ;
            if(l > 0 && dis[e] < dis[tt] + c )
            {
                dis[e] = dis[tt] + c ;
                pre[e] = tt ;
                path[e] = i ;
                if(!vis[e])
                {
                    vis[e] = 1 ;
                    qe[h ++ ] = e ;
                }
            }
        }
    }
    return dis[T] != -1 ;
}
int MFMC()
{
    int mm = 0 ;
    while(spfa())
    {
        int ff = inf ;
        for (int i = T ;i != S ;i = pre[i] )
        ff = min(ff,ed[path[i]].l) ;
        for (int i = T ;i != S ;i = pre[i])
        ed[path[i]].l -= ff ,ed[path[i] ^ 1].l += ff ;
        mm += dis[T] * ff ;
    }
    return mm ;
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("acm.txt", "r", stdin);
#endif
    scanf("%d%d",&n,&k) ;
    init() ;
    REP(i,1,n)
    REP(j,1,n)scanf("%d",&Map[i][j]) ;
    build() ;
    printf("%d\n",MFMC()) ;
    return 0;
}