poj3422

链接：点击打开链接

题意：给一个N*N的矩阵，从(1,1)到(n,n)走K次，每次加上走过的点的权值，然后这个点上面的权值变为0，求走过K次后和的最大值

代码：

#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
    int to,cap,cost,rev;
};
int V;
int dis[10005],prevv[10005],preve[10005];
vector<node> G[10005];
void addedge(int from,int to,int cap,int cost){
    G[from].push_back((node){to,cap,cost,G[to].size()});
    G[to].push_back((node){from,0,-cost,G[from].size()-1});
}
int min_cost_flow(int s,int t,int f){
    int i,d,v,ans,sign;
    ans=0;
    while(f>0){
        fill(dis,dis+V,INF);
        dis[s]=0;
        sign=1;
        while(sign){
        sign=0;
        for(v=0;v<V;v++){
            if(dis[v]==INF)
            continue;
            for(i=0;i<G[v].size();i++){
                node &e=G[v][i];
                if(e.cap>0&&dis[e.to]>dis[v]+e.cost){
                    dis[e.to]=dis[v]+e.cost;
                    prevv[e.to]=v;
                    preve[e.to]=i;
                    sign=1;
                }
            }
        }
        }
        if(dis[t]==INF)
        return -1;
        d=f;
        for(v=t;v!=s;v=prevv[v])
        d=min(d,G[prevv[v]][preve[v]].cap);
        f-=d;
        ans+=d*dis[t];
        for(v=t;v!=s;v=prevv[v]){
            node &e=G[prevv[v]][preve[v]];
            e.cap-=d;
            G[v][e.rev].cap+=d;
        }
    }
    return ans;
}                                               //费用流模板
int s[505][505];
int main(){
    int n,k,i,j,x,y,tmp;
    while(scanf("%d%d",&n,&k)!=EOF){
        V=2*n*n+1;
        for(i=0;i<=n*n+1;i++)
        G[i].clear();
        for(i=0;i<n;i++)
        for(j=0;j<n;j++){
        scanf("%d",&s[i][j]);
        s[i][j]*=-1;
        }                                       //因为是最大值所以取反
        if(k==0){
            puts("0");
            continue;
        }
        addedge(0,1,INF,0);
        for(i=0;i<n;i++)
        for(j=0;j<n;j++){
            tmp=2*(i*n+j)+1;    
            x=2*((i+1)*n+j)+1,y=2*(i*n+j+1)+1;
            addedge(tmp,tmp+1,1,s[i][j]);       //因为权值在点上因此将每个点拆成两个点，一条容量是
            addedge(tmp,tmp+1,INF,0);           //1，费用是该店的值，另一条容量为INF，费用为0，从而
            if(i+1<n)                           //实现一条路走过之后费用为0
            addedge(tmp+1,x,INF,0);
            if(j+1<n)
            addedge(tmp+1,y,INF,0);
        }
        printf("%d\n",-min_cost_flow(0,V-1,k));
    }
    return 0;
}