【子矩阵——优先队列】

题目

代码

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e3 + 10;
const int mod = 998244353;
int g[N][N], ymax[N][N], ymin[N][N];
deque<int> qxmax[N], qxmin[N], qymax[N], qymin[N];int main() {ios::sync_with_stdio(0);cin.tie(0);int n, m, a, b;cin >> n >> m >> a >> b;// 初始化矩阵for (int i = 1; i <= n; i++) {for (int j = 1; j <= m; j++) {cin >> g[i][j];}}// 垂直方向的滑动窗口（按列计算）for (int j = 1; j <= m; j++) {for (int i = 1; i <= n; i++) {// 最大值队列while (!qymax[j].empty() && g[qymax[j].back()][j] <= g[i][j]) qymax[j].pop_back();qymax[j].push_back(i);// 最小值队列while (!qymin[j].empty() && g[qymin[j].back()][j] >= g[i][j]) qymin[j].pop_back();qymin[j].push_back(i);// 窗口内数据有效if (i >= a) {if (i - qymax[j].front() + 1 > a) qymax[j].pop_front();if (i - qymin[j].front() + 1 > a) qymin[j].pop_front();ymax[i][j] = g[qymax[j].front()][j]; // 存储具体值ymin[i][j] = g[qymin[j].front()][j]; // 存储具体值}}}// 水平方向的滑动窗口（按行计算）ll res = 0;for (int i = a; i <= n; i++) {for (int j = 1; j <= m; j++) {// 最大值队列while (!qxmax[i].empty() && ymax[i][qxmax[i].back()] <= ymax[i][j]) qxmax[i].pop_back();qxmax[i].push_back(j);// 最小值队列while (!qxmin[i].empty() && ymin[i][qxmin[i].back()] >= ymin[i][j]) qxmin[i].pop_back();qxmin[i].push_back(j);// 窗口内数据有效if (j >= b) {if (j - qxmax[i].front() + 1 > b) qxmax[i].pop_front();if (j - qxmin[i].front() + 1 > b) qxmin[i].pop_front();res = (res + (ll)ymax[i][qxmax[i].front()] * ymin[i][qxmin[i].front()]) % mod;}}}// 输出结果cout << res << endl;return 0;
}