P2258 子矩阵——搜索+dp

P2258 子矩阵

二进制枚举套二进制枚举能过多一半的点；

我们只需要优化一下第二个二进制枚举的部分；

首先我们先枚举选哪几行，再预处理我们需要的差值，上下，左右；

sum_shang,sum_heng

然后DP查找最小值

dp[i][j]表示前i列已经选了j列；

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=;
int n,m,r,c;
int a[maxn][maxn];
int id[maxn];
int b[maxn];
int sum_s[maxn];
int sum_h[maxn][maxn];

void pre_pare()
{
    memset(sum_s,,sizeof(sum_s));
    int num=;
    for(int i=;i<=n;i++) if(b[i]) id[++num]=i;
    for(int i=;i<r;i++)
    {
        for(int j=;j<=m;j++)
        {
            sum_s[j]+=abs(a[id[i]][j]-a[id[i+]][j]);
        }
    }

    for(int i=;i<=m;i++)
    {
        for(int j=i+;j<=m;j++)
        {
            sum_h[i][j]=;
            for(int k=;k<=r;k++)
            {
                sum_h[i][j]+=abs(a[id[k]][i]-a[id[k]][j]);
            }
        }
    }

}

int ans=,res=;
int dp[maxn][maxn];

int query()
{
    memset(dp,0x3f,sizeof(dp));
    res=;
    for(int i=;i<=m;i++)
    {
        dp[i][]=sum_s[i];
        for(int j=;j<=c;j++)
        {
            for(int k=;k<i;k++)
            {
                dp[i][j]=min(dp[i][j],dp[k][j-]+sum_s[i]+sum_h[k][i]);
            }
        }
        res=min(res,dp[i][c]);
    }
    return res;
}

void dfs(int x,int sum)
{
    if(sum>r) return ;
    if(x==n+)
    {
        if(sum!=r) return ;
        pre_pare();
        ans=min(ans,query());
        return ;
    }
    b[x]=;
    dfs(x+,sum);
    b[x]=;
    dfs(x+,sum+);
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&r,&c);
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    dfs(,);
    printf("%d",ans);
    return ;
}