poj 3422 Kaka's Matrix Travels

题意：给出一个矩阵上的数字，要求从左上走到右下，每次只能向右或向下走取数，被取过的地方不再取。
求走k次的最大和。

这个题的建图很巧妙，把一个点每个点拆成两个点使用，并在他们中间建两条边，一条边的容量为1，费用为该点的权值，另一条边的容量为inf，费用为0，这保证每个点的数只能被取走一次，同时又保证了每个点可以重复走。
再根据题意可以向下走和向右走，每个点（当然要注意除了右边界和下边界上的点）和对应的下面一个点以及右边一个点分别建一条边，容量为inf,费用为0.然后建图求解最大费用流。

#include<queue>
#include<stdio.h>
#include<iostream>
using namespace std;
#define N 20000
#define inf 10000000
struct node{  
int u,v,cost,flag,next;  
}edge[N];  
int pre[N],vis[N],head[N],dis[N];  
int n,k,e,ans,c,s,t,temp;

void add_edge(int u,int v,int f,int c)  
{  
     edge[e].u=u;  edge[e].v=v;  edge[e].cost=c;  edge[e].flag=f;  edge[e].next=head[u];   head[u]=e++;    
     edge[e].u=v;  edge[e].v=u;  edge[e].cost=-c; edge[e].flag=0;  edge[e].next=head[v];   head[v]=e++;  
} 

void spfa()  
{  
     memset(pre,-1,sizeof(pre));  
     memset(vis,0,sizeof(vis));  
     for(int i=s;i<=t;i++)  
        dis[i]=-inf;  
     dis[s]=0;  
     queue<int>q;  
     q.push(s);  
     vis[s]=1;    
     while(!q.empty())  
     {    
        int u=q.front();  
        q.pop();   
        vis[u]=0;  
        for(int i=head[u];i!=-1;i=edge[i].next)  
        {  
            int v=edge[i].v;  
            if(edge[i].flag&&dis[v]<dis[u]+edge[i].cost)  
            {  
                dis[v]=dis[u]+edge[i].cost;  
                pre[v]=i;  
                if(!vis[v])  
                {  
                   q.push(v);  
                   vis[v]=1;  
                }   
            }   
        }  
     }  
}  

int work()  
{  
   for(int i=0;i<k;i++)  
   {  
      spfa();  
      ans+=dis[t];  
      for(int j=pre[t];j!=-1;j=pre[edge[j].u])  
      {  
          edge[j].flag-=1;  
          edge[j^1].flag+=1;  
      }  
   }  
   return ans;  
}  

int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        e=ans=0;
        memset(head,-1,sizeof(head));
        s=0,t=2*n*n+1,temp=n*n;
        for(int i=1;i<=n;i++)
           for(int j=1;j<=n;j++)
           {
              scanf("%d",&c);
              add_edge((i-1)*n+j,(i-1)*n+j+temp,1,c);
              add_edge((i-1)*n+j,(i-1)*n+j+temp,inf,0);
              if(j<n)
                 add_edge((i-1)*n+j+temp,(i-1)*n+j+1,inf,0);
              if(i<n)
                 add_edge((i-1)*n+j+temp,i*n+j,inf,0);
           }
        add_edge(0,1,k,0);
        add_edge(2*n*n,2*n*n+1,k,0);
        printf("%d\n",work());
    }
    return 0;
}