这题上来一看这数据范围,一上来开始就写暴力搜索,先从 n 行里选出 r 行,再从 m 列里选出 c 列,然后等所有都选完之后再生成子矩阵,最后再计算一遍,取最小值,复杂度 O ( 2 n × 2 m ) O(2^n \times 2^m) O(2n×2m) ,由于数据过于水,拿了 60 pts ,代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define rp(i, e) for(int i=1;i<=(e);i++)
#define pr(i, e) for(int i=(e);i>=1;i--)
#define rp0(i, e) for(int i=0;i<(e);i++)
#define pr0(i, e) for(int i=(e-1);i>=0;i--)
#define rps(i, b, e) for(int i=(b);i<=(e);i++)
#define prs(i, e, b) for(int i=(e);i>=(b);i--)
using namespace std;
const int NR=20;
int n, m, r, c, a[NR][NR];
namespace DFS
{
//代码习惯,每一个解法写一个namespace
int ans=2e9, hang[NR], lie[NR], p[NR][NR];
inline int chk()
{
int res=0;
rp(i, r)
rp(j, c)
{
p[i][j]=a[hang[i]][lie[j]];
res+=(i>1)*abs(p[i][j]-p[i-1][j])+
(j>1)*abs(p[i][j]-p[i][j-1]);
}
return res;
}
void dfs_lie(int step, int cnt)
{
if(cnt>c)
{
ans=min(ans, chk());
return;
}
if(step>m)return;
dfs_lie(step+1, cnt);
lie[cnt]=step;
dfs_lie(step+1, cnt+1);
}
void dfs_hang(int step, int cnt)
{
if(cnt>r)
{
dfs_lie(1, 1);
return;
}
if(step>n)return;
dfs_hang(step+1, cnt);
hang[cnt]=step;
dfs_hang(step+1, cnt+1);
}
void solve()
{
dfs_hang(1, 1);
printf("%d", ans);
}
}
int main()
{
scanf("%d%d%d%d", &n, &m, &r, &c);
rp(i, n)rp(j, m)scanf("%d", a[i]+j);
DFS::solve();
return 0;
}
然后我开始思考这个暴力有什么可以优化的地方,然后我发现其实没有必要最后再生成矩阵,其实可以搜索列边计算已有分值,拿了 65 pts ,代码:<