Luogu P2258 [NOIP2014] 子矩阵（枚举+背包dp）

题目链接：https://www.luogu.com.cn/problem/P2258

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=21,MAXX=1<<29;
int n,m,r,c,a[N][N],lj[N],ins[N],dp[N][N],ans;
bool vis[N];

inline int re_ad() {
    char ch=getchar(); int x=0,f=1;
    while(ch<'0' || ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while('0'<=ch && ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;
}

inline int dabs(int d) { return d>0?d:-d; }

inline void init() {
    memset(ins,0,sizeof(ins));
    for(int i=1;i<=n;++i) for(int j=2;j<=c;++j) ins[i]+=dabs(a[i][lj[j]]-a[i][lj[j-1]]);
}

inline int cal(int d,int e) {
    int res=0;
    for(int i=1;i<=c;++i) res+=dabs(a[d][lj[i]]-a[e][lj[i]]);
    return res;
}

inline void print_test() {
    printf("%d\n",ans);
    for(int i=1;i<=c;++i) printf("%d ",lj[i]); puts("");
    for(int i=1;i<=n;++i) printf("%d ",ins[i]); puts("");
    for(int i=1;i<=n;++i) {
        for(int j=1;j<=r;++j) printf("%d ",dp[i][j]);
        puts("");
    }
    puts("");
}

inline void Solve() {
    init();
    memset(dp,0x3f,sizeof(dp));
    for(int i=0;i<=n;++i) dp[i][1]=ins[i];
    for(int i=2;i<=n;++i) for(int j=2;j<=min(i,r);++j) {
        for(int k=1;k<i;++k) dp[i][j]=min(dp[i][j],dp[k][j-1]+cal(k,i)+ins[i]);
    }
    for(int i=r;i<=n;++i) ans=min(ans,dp[i][r]);
//    if(lj[1]==1 && lj[2]==3 && lj[3]==4) print_test();
}

void search(int d,int x) {
    if(d>c) { Solve(); return ; }
    for(int i=x;i<=m;++i) if(!vis[i]) vis[i]=true,lj[d]=i,search(d+1,i+1),vis[i]=false;
}

int main()
{
    n=re_ad(),m=re_ad(),r=re_ad(),c=re_ad();
    for(int i=1;i<=n;++i) for(int j=1;j<=m;++j) a[i][j]=re_ad();
    ans=MAXX;
    search(1,1);
    printf("%d\n",ans);
    return 0;
}