洛谷 - P2045 - 方格取数加强版 - 费用流

原来这种题的解法是费用流。

从一个方格的左上走到右下，最多走k次，每个数最多拿走一次。

每次走动的流量设为1，起始点拆点成限制流量k。

每个点拆成两条路，一条路限制流量1，费用为价值相反数。另一条路无限流量。

跑一遍费用流。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=700005+1;
const int MAXM=700000;
const int INF=0x3f3f3f3f;
struct Edge{
    int to,next,cap,flow,cost;
}edge[MAXM];
int head[MAXN],tol;
int pre[MAXN],dis[MAXN];
bool vis[MAXN];

int n;
void init(){
    tol=0;
    memset(head,-1,sizeof(head));
}

void addedge(int u,int v,int cap,int cost){
    edge[tol].to=v;
    edge[tol].cap=cap;
    edge[tol].cost=cost;
    edge[tol].flow=0;
    edge[tol].next=head[u];
    head[u]=tol++;

    edge[tol].to=u;
    edge[tol].cap=0;
    edge[tol].cost=-cost;
    edge[tol].flow=0;
    edge[tol].next=head[v];
    head[v]=tol++;
}

bool spfa(int s,int t){
    queue<int> q;
    memset(dis,INF,sizeof(dis));
    memset(vis,false,sizeof(vis));
    memset(pre,-1,sizeof(pre));

    dis[s]=0;
    vis[s]=true;
    q.push(s);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].to;
            if(edge[i].cap>edge[i].flow&&dis[v]>dis[u]+edge[i].cost){
                dis[v]=dis[u]+edge[i].cost;
                pre[v]=i;
                if(!vis[v]){
                    vis[v]=true;
                    q.push(v);
                }
            }
        }
    }
    if(pre[t]==-1)
        return false;
    else
        return true;
}

int minCostMaxFlow(int s,int t,int &cost){
    int flow=0;
    cost=0;
    while(spfa(s,t)){
        int Min=INF;
        for(int i=pre[t];i!=-1;i=pre[edge[i^1].to]){
            if(Min>edge[i].cap-edge[i].flow)
                Min=edge[i].cap-edge[i].flow;
        }
        for(int i=pre[t];i!=-1;i=pre[edge[i^1].to]){
            edge[i].flow+=Min;
            edge[i^1].flow-=Min;
            cost+=edge[i].cost*Min;
        }
        flow+=Min;
    }
    return flow;
}


void show(int s){
    bool vis[20];
    queue<int> q;
    vis[s]=1;
    q.push(s);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        cout<<"u="<<u<<endl;
        for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].to;
            if(vis[v]==0){
                vis[v]=1;
                q.push(v);
            }
        }
    }
    cout<<endl;
}

/* EK end */

int a[55][55];

int k;

inline int getid(int i,int j,int isout){
    return (i-1)*n+j+isout*(n*n);
}

int main(){
    init();
    scanf("%d%d",&n,&k);

    int si=2*n*n+1,so=2*n*n+2;
    addedge(si,so,k,0);//最多走k次
    int t=2*n*n+3;

    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int c;
            scanf("%d",&c);
            addedge(getid(i,j,0),getid(i,j,1),1,-c);
            //拿走它，获得价值，费用是相反数
            addedge(getid(i,j,0),getid(i,j,1),INF,0);
            //不拿就不拿呗
        }
    }

    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i+1<=n)
                addedge(getid(i,j,1),getid(i+1,j,0),INF,0);
            if(j+1<=n)
                addedge(getid(i,j,1),getid(i,j+1,0),INF,0);
        }
    }

    addedge(so,getid(1,1,0),INF,0);
    addedge(getid(n,n,1),t,INF,0);

    //show(si);

    int cost=0;
    int flow=minCostMaxFlow(si,t,cost);
    printf("%d\n",-cost);

}