HDU - 2888 Check Corners 【RMQ（二维线段树 / ST）】

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2888

第一种方法是二维线段树，注意的地方在代码注释的部分。第二种方法是 ST算法，在后面。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long ll;

const int Maxn = 300+10;
const int INF = 0x3f3f3f3f;

struct Node {
    int maxn;
}node[Maxn<<2][Maxn<<2];

int a[Maxn][Maxn];

void subupdata(int c1, int c2, int l, int r, int L, int R) {
    if(l == r && L == R) {
        node[c1][c2].maxn = a[l][L]; return;
    }
    if(L == R) {
        node[c1][c2].maxn = max(node[c1<<1][c2].maxn, node[c1<<1|1][c2].maxn); return;
    }
    int mid = (L+R)/2;
    subupdata(c1, c2<<1, l, r, L, mid);
    subupdata(c1, c2<<1|1, l, r, mid+1, R);
    node[c1][c2].maxn = max(node[c1][c2<<1].maxn, node[c1][c2<<1|1].maxn);
}

void updata(int c1, int c2, int l, int r, int L, int R) {
    int mid = (l+r)/2;
    if(l != r) {
        updata(c1<<1, c2, l, mid, L, R);
        updata(c1<<1|1, c2, mid+1, r, L, R);
    }
    subupdata(c1, c2, l, r, L, R);
}

int x_1, x_2, y_1, y_2; // 参数要拿出来，如果带到query里面递归会超内存

int subquery(int c1, int c2, int L, int R) {
    if(y_1 <= L && R <= y_2) return node[c1][c2].maxn;
    int mid = (L+R)/2, maxn = -INF;
    if(y_1 <= mid) maxn = max(maxn, subquery(c1, c2<<1, L, mid));
    if(mid+1 <= y_2) maxn = max(maxn, subquery(c1, c2<<1|1, mid+1, R));
    return maxn;
}

int query(int c1, int c2, int l, int r, int L, int R) {
    if(x_1 <= l && r <= x_2) return subquery(c1, c2, L, R);
    int mid = (l+r)/2, maxn = -INF;
    if(x_1 <= mid) maxn = max(maxn, query(c1<<1, c2, l, mid, L, R));
    if(mid+1 <= x_2) maxn = max(maxn, query(c1<<1|1, c2, mid+1, r, L, R));
    return maxn;
}

int main(void)
{
    int N, M;
    while(scanf("%d%d", &N, &M) != EOF) {
        for(int i = 1; i <= N; ++i) {
            for(int j = 1; j <= M; ++j) scanf("%d", &a[i][j]);
        }
        updata(1, 1, 1, N, 1, M);
        int q, maxn;
        scanf("%d", &q);
        while(q--) {
            scanf("%d%d%d%d", &x_1, &y_1, &x_2, &y_2);
            maxn = query(1, 1, 1, N, 1, M);
            printf("%d ", maxn);
            if(a[x_1][y_1] == maxn || a[x_1][y_2] == maxn || a[x_2][y_1] == maxn || a[x_2][y_2] == maxn) printf("yes\n");
            else printf("no\n");
        }
    }
    return 0;
}

 

 二维的ST要比一维的麻烦一些，这里主要是把代码dp的下标讲一下。（下面的内容是针对已经了解一维ST的朋友）

dp[ i ][ j ][ k ][ l ]。

首先是 k，l 这两个下标表示的是二维数组对应的位置，i 和 j 分别表示 行的间隔和列的间隔。

假设 i == 0， j == 1，然后遍历k，l，代表的意思就是，以k，l为起点计算 1 (2^0) 行内，2 (2^1) 列内的最大值，当遍历完所有的坐标k，l之后，那么任意起点的一行两列内的最大值都求出来了，随后求 i == 0, j == 2的最大值，这里要区分的是 i == 0 和 i != 0的情况，当 i > 0 时，说明任意起点任意列的间隔的最大值都已经求出来了，那么对于 i > 0 的情况，只需要把 i -1 的情况拿出来比较最大值就行了，j 的情况在 i == 0的时候就已经求出来了。

在query的部分要注意的是，需要比较的最大值有四部分，和一维的不同(只需要两部分，第一部分覆盖不到的地方由第二部分补上)，这里二维需要用四块补上整个矩阵。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

typedef long long ll;

const int Maxn = 300+10;
const int INF = 0x3f3f3f3f;

int dp[9][9][Maxn][Maxn];

void st(int n, int m) {
    memset(dp, -1, sizeof(dp));
    for(int i = 1; i <= n; ++i) {
        for(int j = 1; j <= m; ++j) scanf("%d", &dp[0][0][i][j]);
    }
    for(int i = 0; (1<<i) <= n; ++i) {
        for(int j = 0; (1<<j) <= m; ++j) {
            if(!i && !j) continue;
            for(int k = 1; k+(1<<i)-1 <= n; ++k) {
                for(int l = 1; l+(1<<j)-1 <= m; ++l) {
                    if(i == 0) dp[i][j][k][l] = max(dp[i][j-1][k][l], dp[i][j-1][k][l+(1<<(j-1))]);
                    else {
                        dp[i][j][k][l] = max(dp[i-1][j][k][l], dp[i-1][j][k+(1<<(i-1))][l]);
                    }
                }
            }
        }
    }
}

int query(int x1, int y1, int x2, int y2) {
    int i = (int)((double)log(x2-x1+1)/log(2.0));
    int j = (int)((double)log(y2-y1+1)/log(2.0));
    return max(max(dp[i][j][x1][y1], dp[i][j][x1][y2-(1<<j)+1]),
               max(dp[i][j][x2-(1<<i)+1][y1], dp[i][j][x2-(1<<i)+1][y2-(1<<j)+1]));
}

int main(void)
{
    int N, M;
    while(scanf("%d%d", &N, &M) != EOF) {
        st(N, M);
        int q, x1, x2, y1, y2;
        scanf("%d", &q);
        while(q--) {
            scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
            int maxn = query(x1, y1, x2, y2);
            printf("%d ", maxn);
            if(dp[0][0][x1][y1] == maxn || dp[0][0][x1][y2] == maxn || dp[0][0][x2][y1] == maxn || dp[0][0][x2][y2] == maxn)
                printf("yes\n");
            else printf("no\n");
        }
    }
    return 0;
}