bzoj5359 [Lydsy1805月赛]寻宝游戏（dp+贪心）

假设已经选定了一条路线，那么不考虑具体交换方案，考虑选定一个$t(t\leq k)$，经过的格子里要有t 个格子的权值不计入得分，而不经过的格子里要有t 个格子的权值计入总分。那么每一种交换方案都可以对应这样一个转化。
设$f_{i,j,x,y}$表示从$(1,1)$ 出发来到$(i,j)$，考虑完前$i-1$行所有格子以及第i 行前j 个格子时，有x 个经过的格子不计分，y 个不经过的格子计分的情况下，总分的最大值是多少。
那么有两种状态转移：
• 往右走一格，直接转移到$f_{i,j+1,x,y},f_{i,j+1,x+1,y}$
• 往下走一格，转移到$f_{i+1,j,x′,y′}$，需要枚举这一行及下一行有多少个不经过的格子计分。显然这些格子一定按照权值从大到小贪心选择。
最终答案即为$max\{f_{n,m,t,t}\}(0\leq t\leq k)$，时间复杂度$O(n^2(nlogn+k^3))$。

注意初始化要给-inf,因为f[1][1][n][0]这种东西是不合法的。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 60
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,K,a[N][N],f[N][N][22][22],b[N];
//f[i][j][x][y],走到了i,j,考虑了前i-1行及第i行前j列,路线上x个不选,不在路线上的y个选的最大权值和 
inline bool cmp(int x,int y){return x>y;}
inline void Max(int &x,int y){x=max(x,y);}
int main(){
//  freopen("a.in","r",stdin);
    int tst=read();
    while(tst--){
        n=read();m=read();K=read();memset(f,128,sizeof(f));
        for(int i=1;i<=n;++i)
            for(int j=1;j<=m;++j) a[i][j]=read();
        f[1][1][0][0]=a[1][1];f[1][1][1][0]=0;
        for(int i=1;i<=n;++i)
            for(int j=1;j<=m;++j){
                if(j<m){
                    for(int x=0;x<=K;++x)
                        for(int y=0;y<=K;++y){
                            Max(f[i][j+1][x][y],f[i][j][x][y]+a[i][j+1]);
                            if(x<K) Max(f[i][j+1][x+1][y],f[i][j][x][y]);
                        }
                }if(i==n) continue;int nn=0;
                for(int k=j+1;k<=m;++k) b[++nn]=a[i][k];
                for(int k=1;k<j;++k) b[++nn]=a[i+1][k];
                sort(b+1,b+nn+1,cmp);
                for(int k=1;k<=nn;++k) b[k]+=b[k-1];
                for(int x=0;x<=K;++x)
                    for(int y=0;y<=K;++y)
                        for(int k=0;k<=nn;++k){
                            if(y+k>K) break;
                            Max(f[i+1][j][x][y+k],f[i][j][x][y]+a[i+1][j]+b[k]);
                            if(x<K) Max(f[i+1][j][x+1][y+k],f[i][j][x][y]+b[k]);
                        }
            }
        int ans=0;for(int k=0;k<=K;++k) Max(ans,f[n][m][k][k]);
        printf("%d\n",ans);
    }return 0;
}