(dp/单调队列、st表优化)P2216 [HAOI2007]理想的正方形

https://www.luogu.org/problemnew/show/P2216#sub
有一个a*b的整数组成的矩阵,现请你从中找出一个n*n的正方形区域,使得该区域所有数中的最大值和最小值的差最小。
in:
5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2
out:
1

从a * b的矩阵里选出一个n * n的矩阵,使矩阵内最大值和最小值差值最小。
我们可以想到分别求出矩阵的最小值和最大值求解
我们用dp[k][i][j][2]表示以i, j为左上角大小为k * k的矩阵的最大最小值
对于左上角(i, j)的矩阵,他的最值即是dp[k][i][j][0/1] = min/max(dp[k - 1][i][j][0/1], min(dp[k - 1][i][j + 1][0/1], min(dp[k - 1][i + 1][j][0/1], dp[k - 1][i + 1][j + 1][0/1])));
但程序空间使用太大,直接优化掉第一维,可以循环直接求解
而时间复杂度也不优秀,在O2优化后可以过,应该要用单调队列优化

#include
using namespace std;
typedef long long ll;
const int maxn = 1000 + 5;
inline void read(int &x){
    int data = 0, w = 1;
    char ch = getchar();
    while(ch != '-' && !isdigit(ch))
        ch = getchar();
    if(ch == '-')
        w = -1, ch = getchar();
    while(isdigit(ch))
        data = 10 * data + ch - '0', ch = getchar();
    x = data * w;
}
void write(int x){
    if(x < 0)
        putchar('-'), x = -x;
    if(x > 9)
        write(x / 10);
    putchar('0' + x % 10);
}
int x[maxn][maxn];
int dp[maxn][maxn][2]; //max - min
int main()
{
    int a, b, n;
    read(a), read(b), read(n);
    for(int i = 1; i <= a; i++)
        for(int j = 1; j <= b; j++) {
            read(x[i][j]);
            dp[i][j][0] = dp[i][j][1] = x[i][j];
        }
    for(int k = 2; k <= n; k++) {
        for(int i = 1; i <= a; i++) {
            for(int j = 1; j <= b; j++) {
                if(k + i - 1 > a || j + k - 1 > b) continue;
                dp[i][j][0] = min(dp[i][j][0], min(dp[i][j + 1][0], min(dp[i + 1][j][0], dp[i + 1][j + 1][0])));
                dp[i][j][1] = max(dp[i][j][1], max(dp[i][j + 1][1], max(dp[i + 1][j][1], dp[i + 1][j + 1][1])));
            }
        }
    }
    int ans = 0x3f3f3f3f;
    for(int i = 1; i <= a; i++) {
        for(int j = 1; j <= b; j++) {
            if(i + n - 1 > a || j + n - 1 > b) continue;
            ans = min(ans, dp[i][j][1] - dp[i][j][0]);
        }
    }
    write(ans);
    return 0;
}

优化可以用st表或者单调队列
正如上面所说,每个矩阵的值由它本身、右矩阵、下矩阵、右下矩阵的上一状态的值决定,发现和st表的原理很像,可以写成二维的st表
st[i][j][2]表示由(i,j)为左上角,大小为k * k的矩阵的最大值和最小值,节省空间直接优化掉第一维k
所以每个状态转移为st[i][j][0/1] = min/max(st[i][j][0/1], st[i+(1<

#include
using namespace std;

const int maxn = 1e3 + 100;
int arr[maxn][maxn], len, n;
int st[maxn][maxn][2];
int query(int i, int j) {
    int _max, _min;
    _max = max(st[i][j][1], max(st[i + n - (1 << len)][j][1], max(st[i][j + n - (1 << len)][1], st[i + n - (1 << len)][j + n - (1 << len)][1])));
    _min = min(st[i][j][0], min(st[i + n - (1 << len)][j][0], min(st[i][j + n - (1 << len)][0], st[i + n - (1 << len)][j + n - (1 << len)][0])));
    return _max - _min;

}
int main()
{
    int a, b;
    cin >> a >> b >> n;
    for(int i = 1; i <= a; i++)
        for(int j = 1; j <= b; j++)
            cin >> arr[i][j], st[i][j][0] = st[i][j][1] = arr[i][j];
    //预处理st表
    len = 0;
    while((1 << (len + 1)) <= n) len++;
    for(int k = 0; k < len; k++) {
        for(int i = 1; i + (1 << k) <= a; i++) {
            for(int j = 1; j + (1 << k) <= b; j++) {
                st[i][j][0] = min(st[i][j][0], min(st[i + (1 << k)][j][0], min(st[i][j + (1 << k)][0], st[i + (1 << k)][j + (1 << k)][0])));
                st[i][j][1] = max(st[i][j][1], max(st[i + (1 << k)][j][1], max(st[i][j + (1 << k)][1], st[i + (1 << k)][j + (1 << k)][1])));
            }
        }
    }
    维护由i,j为左上角的大小为n的矩阵值
    int ans = 0x3f3f3f3f;
    for(int i = 1; i <= a - n + 1; i++) {
        for(int j = 1; j <= b - n + 1; j++) {
            ans = min(ans, query(i, j));
        }
    }
    cout << ans << endl;
}

你可能感兴趣的:(总结)