P2258 子矩阵（DP）

题目链接

题面：

题解：
枚举行然后 dp 列。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=30;
int n,m,r,c;
int a[maxn][maxn],ha[maxn],dp[maxn][maxn];
int l[maxn],h[maxn][maxn];
int ans=inf;

void init(void)
{
    memset(l,0,sizeof(l));
    memset(h,0,sizeof(h));
    //枚举如果选了这一列列上的贡献
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<r;j++)
            l[i]+=abs(a[ha[j]][i]-a[ha[j+1]][i]);
    }

    //枚举 选了i，j两列且中间没有其他列被选，行上的贡献
    for(int i=1;i<=m;i++)
    {
        for(int j=i+1;j<=m;j++)
        {
            for(int k=1;k<=r;k++)
                h[i][j]+=abs(a[ha[k]][i]-a[ha[k]][j]);
        }
    }

}

int DP(void)
{
    memset(dp,0x3f,sizeof(dp));
    int res=inf;
    //dp[i][j] 选第i列为最后一列，且一共选了j列的贡献
    for(int i=1;i<=m;i++)
    {
        dp[i][1]=l[i];
        for(int j=2;j<=c;j++)
        {
            for(int k=1;k<i;k++)
                dp[i][j]=min(dp[i][j],dp[k][j-1]+l[i]+h[k][i]);
        }
        res=min(res,dp[i][c]);
    }
    return res;
}

//枚举选了哪些行
void dfs(int cnt)
{
    if(cnt==r+1)
    {
        init();
        ans=min(ans,DP());
        return ;
    }
    for(int i=ha[cnt-1]+1;i<=n;i++)
    {
        ha[cnt]=i;
        dfs(cnt+1);
        ha[cnt]=0;
    }
}

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