[算法题解] acwing 4964.子矩阵

题目描述

给定一个 n×m
（n
行 m
列）的矩阵。

设一个矩阵的价值为其所有数中的最大值和最小值的乘积。

求给定矩阵的所有大小为 a×b
（a
行 b
列）的子矩阵的价值的和。

答案可能很大，你只需要输出答案对 998244353
取模后的结果。

输入格式
输入的第一行包含四个整数分别表示 n,m,a,b
，相邻整数之间使用一个空格分隔。

接下来 n
行每行包含 m
个整数，相邻整数之间使用一个空格分隔，表示矩阵中的每个数 Ai,j
。

输出格式
输出一行包含一个整数表示答案。

数据范围
对于 40%
的评测用例，1≤n,m≤100
；
对于 70%
的评测用例，1≤n,m≤500
；
对于所有评测用例，1≤a≤n≤1000
，1≤b≤m≤1000
，1≤Ai,j≤109
。

样例

输入样例：
2 3 1 2
1 2 3
4 5 6
输出样例：
58
样例解释
1×2+2×3+4×5+5×6=58
。

思路：求一个二维空间里滑动窗口(a x b)中的最值，可以联想到使用单调队列.先固定左右边界，用单调队列分
别求出a行长度为b的滑动窗口的最大值和最小值，并将求的的最值保存到数组中现在就转化为了求一维空间滑动窗口的最小值。

总结：二维空间滑动窗口的最值问题可转化为求两次一维空间滑动窗口的最值（用两次单调队列）

---

C++ 代码 $O(mn)$

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 1010;
const int mod = 998244353;
int num[N][N];
int rmax[N][N],rmin[N][N];
int q[N];
void get_max(int a[],int b[],int x,int y){
    int hh = 0,tt = -1;
    for(int i = 0;i < x;i ++){
        if(hh <= tt && q[hh] <= i - y) hh ++;
        while(hh <= tt && a[q[tt]] <= a[i]) tt --;
        q[++ tt] = i;
        b[i] = a[q[hh]];
    }
}

void get_min(int a[],int b[],int x,int y){
    int hh = 0,tt = -1;
    for(int i = 0;i < x;i ++){
        if(hh <= tt && q[hh] <= i - y) hh ++;
        while(hh <= tt && a[q[tt]] >= a[i]) tt --;
        q[++ tt] = i;
        b[i] = a[q[hh]];
    }
}
int main()
{
    int n,m,a,b;
    cin >> n >> m >> a >> b;
    for(int i = 0;i < n;i ++){
        for(int j = 0;j < m;j ++){
            scanf("%d",&num[i][j]);
        }
    }
    for(int i = 0;i < n;i ++){
        get_max(num[i],rmax[i],m,b);
        get_min(num[i],rmin[i],m,b);
    }
    int res = 0;
    int e[N],c[N],d[N];
    for(int i = b - 1;i < m;i ++){
        for(int j = 0;j < n;j ++) e[j] = rmax[j][i];
        get_max(e,c,n,a);
        for(int j = 0;j < n;j ++) e[j] = rmin[j][i];
        get_min(e,d,n,a);
        for(int j = a - 1;j < n;j ++){
            res = (res + (long long)c[j] * d[j]) % mod;
        }
    }
    cout << res;
    return 0;
}