P2216 [HAOI2007]理想的正方形
这道题就是标准的二维RMQ模板。
回顾一下原来的RMQ,分两个阶段,先倍增的初始化,再\(O(1)\)地求答案。二维RMQ也是有异曲同工之妙的。
这个最值怎么定义?我们设\(maxv[i][j][k]\)为以\((i,j)\)为左上角的\(2^k\times 2^k\)正方形区域中的最大值。最小值同理,下面都只讨论最大值。
怎么初始化?我们同样枚举\(k\)。一维RMQ通过枚举\(k\)变成两段,二维RMQ就会变成四个。都取最大值就是了。
怎么求答案?同样\(O(1)\)地算出\(n\)对应的\(k\),然后同样取四个ST表得到最大值。
基本原理跟一维RMQ是完全一致的,唯一的不同就是需要讨论4种,仅此而已。
好好看代码8:
#include<bits/stdc++.h>
const int maxn = 1002;
const int INF = 0x3f3f3f3f;
int maxv[12][maxn][maxn], minv[12][maxn][maxn];
int a, b, n;
int ans = INF;
int main() {
scanf("%d %d %d", &a, &b, &n);
for(int i = 1; i <= a; i++) {
for(int j = 1; j <= b; j++) {
scanf("%d", &maxv[0][i][j]);
minv[0][i][j] = maxv[0][i][j];
}
}
for(int k = 1; (1 << k) <= n; k++) {
for(int i = 1; i + (1 << k) - 1 <= a; i++) {
for(int j = 1; j + (1 << k) - 1 <= b; j++) {
maxv[k][i][j] = std::max(std::max(maxv[k - 1][i][j], maxv[k - 1][i + (1 << (k - 1))][j]), std::max(maxv[k - 1][i][j + (1 << (k - 1))], maxv[k - 1][i + (1 << (k - 1))][j + (1 << (k - 1))]));
minv[k][i][j] = std::min(std::min(minv[k - 1][i][j], minv[k - 1][i + (1 << (k - 1))][j]), std::min(minv[k - 1][i][j + (1 << (k - 1))], minv[k - 1][i + (1 << (k - 1))][j + (1 << (k - 1))]));
}
}
}
int k = 0;
while((1 << (k + 1)) <= n) k++;
for(int i = 1; i + n - 1 <= a; i++) {
for(int j = 1; j + n - 1 <= b; j++) {
int max = std::max(std::max(maxv[k][i][j], maxv[k][i + n - (1 << k)][j]), std::max(maxv[k][i][j + n - (1 << k)], maxv[k][i + n - (1 << k)][j + n - (1 << k)]));
int min = std::min(std::min(minv[k][i][j], minv[k][i + n - (1 << k)][j]), std::min(minv[k][i][j + n - (1 << k)], minv[k][i + n - (1 << k)][j + n - (1 << k)]));
ans = std::min(ans, max - min);
}
}
printf("%d\n", ans);
}
参考文献:https://www.cnblogs.com/GXZlegend/p/7491533.html
这篇博客的代码真的漂亮。
漂亮警告