poj3422K方格取数——最大费用最大流

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

最大费用最大流：

拆点，在自点之间连两条边，一条容量为1，边权为数字；一条容量为k-1，边权为0；表示可以走k次，只有一次能取到数字；

从每个格子能向下或向右走，于是向下面的点、右面的点连边；

将E-K算法中的bfs改成spfa求最长路(边权最大)，找出最大流即为答案；

因为对E-K算法不太熟练，WA了n遍，最后发现是不能像dinic一样在bfs中途找到汇点就退出，因为不是构建分层图，而是spfa求最长路。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
queue<int>q;
int n,k,ans,head[],ct=,s,t,inf=1e9,dis[],pre[],incf[];
bool vis[];
struct N{
    int to,next,w,v;
    N(int t=,int n=,int ww=,int vv=):to(t),next(n),w(ww),v(vv) {}
}edge[];
void add(int x,int y,int ww,int vv)
{
    edge[++ct]=N(y,head[x],ww,vv);head[x]=ct;
    edge[++ct]=N(x,head[y],,-vv);head[y]=ct;
}
bool bfs()
{
    memset(vis,,sizeof vis);
    memset(pre,,sizeof pre);
    memset(dis,-,sizeof dis);
    memset(incf,,sizeof incf);
    while(q.size())q.pop();
    q.push(s);vis[s]=;dis[s]=;
    while(q.size())
    {
        int x=q.front();q.pop();vis[x]=;
        for(int i=head[x];i;i=edge[i].next)
        {
            int u=edge[i].to;
//            if(vis[u])continue;
            if(edge[i].w&&dis[u]<dis[x]+edge[i].v)//残量网络spfa
            {
                dis[u]=dis[x]+edge[i].v;
                incf[u]=min(edge[i].w,incf[x]);
                pre[u]=i;
                if(!vis[u])q.push(u),vis[u]=;
//                if(u==t)return 1;//!!!
            }
        }
    }
    if(dis[t]<)return ;
    else return ;
//    return 0;
}
void up()
{
    int x=t;
    while(x!=s)
    {
        int i=pre[x];
        edge[i].w-=incf[t];
        edge[i^].w+=incf[t];
        ans+=edge[i].v*incf[t];//
        x=edge[i^].to;
    }
//    ans+=dis[t]*incf[t];// 皆可
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=;i<=n;i++)
        for(int j=;j<=n;j++)
        {
            int x,p=(i-)*n+j,p2=p+n*n;
            scanf("%d",&x);
            add(p,p2,,x);
            add(p,p2,k-,);//
            if(i<n)add(p2,p+n,k,);
            if(j<n)add(p2,p+,k,);
        }
    s=;t=*n*n+;
    add(s,,k,);
    add(*n*n,t,k,);
    while(bfs())up();
    printf("%d",ans);
    return ;
}

附数据：

数据