题目传送门
一、题目描述
给定一个 \(8×8\) 的棋盘，棋盘的每个小方格都有一个权值 \(w_x,w_y\)
每次我们可以对棋盘进行一次横或竖切，将棋盘分成两块矩形的子棋盘
分割完一次后，我们可以选择两个子棋盘中的一个再继续递归操作。
可以发现题目中给的这个图片，右边这个并不是递归下去做的：他对第一次分割的两个矩形又分别进行了分割（题目要求我们只能保留一个继续分割）
现需要把棋盘按照上述分割方案，分成 \(n\) 块（\(n−1\) 次划分操作）
求一个划分方案，使得各子棋盘的 总分的均方差最小
二、分析
不难发现，递归操作会有很多冗余的重复计算，于是我们可以采用 记忆化搜索 进行优化
\(f[x_1,y_1,x_2,y_2,k]\) 表示这个区间在剩余\(k\)刀的情况下，可以获取到的计算公式最大值
三、代码实现
#include <bits/stdc++.h>
using namespace std;
const int N = 10; //8*8个格子，我们从下标1开始放入，需要用到下标8，开9个。
const int INF = 0x3f3f3f3f;
int n;
int m = 8;
int s[N][N]; //二维前缀和
double f[N][N][N][N][N]; //DP结果数组
double X; //平均值
//二维前缀和
int get_sum(int x1, int y1, int x2, int y2) {
return s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
}
//均方差公式
double get(int x1, int y1, int x2, int y2) {
double sum = get_sum(x1, y1, x2, y2) - X;
return sum * sum / n;
}
/**
* 功能：记忆化搜索
* @param x1 左上角x坐标
* @param y1 左上角y坐标
* @param x2 右下角x坐标
* @param y2 右下角y坐标
* @param k 剩余的刀数
* @return 根据公式计算出的最小值
*/
double dfs(int x1, int y1, int x2, int y2, int k) {
//用一个v来简写一下，要不太长了
double &v = f[x1][y1][x2][y2][k];//这个引用方法很棒,get到了一个知识点
if (v >= 0) return v; //计算过了，就直接返回，不再重复计算
if (k == 0) return v = get(x1, y1, x2, y2);//如果k=1,表示刀都用完了，最终这一块可以计算出来了
//初始化为正无穷
v = INF;
//选择横着切,注意这里i从x1开始,表示切分时上部分有1，2，3... x2-x1+1 行
for (int i = x1; i < x2; i++) {
//放弃上半部分(死值)，选择下半部分(记忆化搜索)
v = min(v, get(x1, y1, i, y2) + dfs(i + 1, y1, x2, y2, k - 1));
//放弃下半部分(死值)，选择上半部分(记忆化搜索)
v = min(v, get(i + 1, y1, x2, y2) + dfs(x1, y1, i, y2, k - 1));
}
//选择纵着切
for (int i = y1; i < y2; i++) {
//放弃左半部分，选择右半部分
v = min(v, get(x1, y1, x2, i) + dfs(x1, i + 1, x2, y2, k - 1));
//放弃右半部分，选择左半部分
v = min(v, get(x1, i + 1, x2, y2) + dfs(x1, y1, x2, i, k - 1));
}
//返回打擂台的最小值
return v;
}
int main() {
cin >> n;
for (int i = 1; i <= m; i++)
for (int j = 1; j <= m; j++) {
//原数组不用保存，直接用一个二维前缀和数组s即可
cin >> s[i][j];
//二维前缀和构建
s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
}
//利用二维前缀和的结果，计算出平均值，注意要使用double的类型转换，防止丢失精度
X = (double) s[m][m] / n;
//将DP数组初始化为负无穷，计算过的>=0 (因为均方差可能为0)，未计算过的为-INF
//方便获取哪个位置是否计算过
memset(f, -0x3f, sizeof f);
//记忆化搜索:因为最后需要切出n块矩形棋盘，其实就是需要切n-1刀，开始dfs模拟切每一刀
printf("%.3lf\n", sqrt(dfs(1, 1, 8, 8, n - 1)));
return 0;
}
