这题和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;
}