[POJ3422]Kaka's Matrix Travels 做题笔记

题目链接：http://poj.org/problem?id=3422

费用流。
做这题的收获是inf边的另一个用处，棋盘的每个点拆开后，拆开的两个点之间连两条边，一条容量1费用为点权值的相反数（负权是因为跑spfa最大流时要用到），表示只能走一次并获得相应分数，另一条容量inf费用为0，表示可以走很多很多次但是没法获得分数。新建超级源S，S向[1,1]连一条容量为k权值为零的边，跑一遍费用流即可。

PS：敲这题的时候拆点建边马虎了一下，结果调了20分钟，下次建这种比较奇葩的图的时候一定要精神集中，尤其是从一个点的“出点”连向另一个店的“入点”时一定要注意。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N=6000,M=50000,inf=0x3fffffff;
int head[N],ver[M<<1],e[M<<1],c[M<<1],next[M<<1];
int d[N],pos[N],pre[N];bool vis[N],vv[N];
int s,t,k,num,tot=1;
int map[100][100];

void add (int u,int v,int w,int cost) {
    ver[++tot]=v;e[tot]=w;c[tot]=cost;next[tot]=head[u];head[u]=tot;
    ver[++tot]=u;e[tot]=0;c[tot]=-cost;next[tot]=head[v];head[v]=tot;
}

bool spfa () {
    queue<int> q;
    memset(vis,0,sizeof(vis));
    memset(pre,-1,sizeof(pre));
    memset(pos,0,sizeof(pos));
    for (int i=0;i<N;i++) d[i]=0x3fffffff;
    q.push(s); d[s]=0; vis[s]=1; pre[s]=s;
    while (!q.empty()) {
        int x=q.front(); q.pop();
        for (int i=head[x];i;i=next[i]) 
            if (e[i]&&d[x]+c[i]<d[ver[i]]) {
                d[ver[i]]=d[x]+c[i];
                pre[ver[i]]=x,pos[ver[i]]=i;
                if (!vis[ver[i]]) q.push(ver[i]),vis[ver[i]]=1;
            }
        vis[x]=0;
    }
    return pre[t]!=-1;
}

int mcf () {
    int cost=0,maxflow=0,aug;
    while (spfa()) {
        aug=0x3fffffff;
        for (int i=t;i!=s;i=pre[i])  
            aug=min(aug,e[pos[i]]);
        maxflow+=aug,cost+=aug*d[t];
        for (int i=t;i!=s;i=pre[i]) 
            e[pos[i]]-=aug,e[pos[i]^1]+=aug;
    }
    return cost;
}

int main () {
    scanf("%d%d",&num,&k);
    for (int i=1;i<=num;i++) 
        for (int j=1;j<=num;j++) {
            scanf("%d",&map[i][j]);
            add(num*(i-1)+j,num*(i-1)+j+num*num,1,-map[i][j]);
            add(num*(i-1)+j,num*(i-1)+j+num*num,inf,0);
        }
    for (int i=1;i<=num;i++) 
        for (int j=1;j<=num-1;j++) 
            add(num*(i-1)+j+num*num,num*(i-1)+j+1,inf,0);
    for (int j=1;j<=num;j++) 
        for (int i=1;i<=num-1;i++) 
            add(num*(i-1)+j+num*num,num*i+j,inf,0);
    s=0,t=2*num*num;
    add(s,1,k,0);
    printf("%d\n",-mcf());

}