【子矩阵——优先队列】

题目

代码

cpp 复制代码
#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;
}
相关推荐
小小小新人121231 小时前
C语言 ATM (4)
c语言·开发语言·算法
你的冰西瓜1 小时前
C++排序算法全解析(加强版)
c++·算法·排序算法
এ᭄画画的北北2 小时前
力扣-31.下一个排列
算法·leetcode
绝无仅有3 小时前
企微审批对接错误与解决方案
后端·算法·架构
用户5040827858393 小时前
1. RAG 权威指南:从本地实现到生产级优化的全面实践
算法
Python×CATIA工业智造5 小时前
详细页智能解析算法:洞悉海量页面数据的核心技术
爬虫·算法·pycharm
无聊的小坏坏5 小时前
力扣 239 题:滑动窗口最大值的两种高效解法
c++·算法·leetcode
黎明smaly5 小时前
【排序】插入排序
c语言·开发语言·数据结构·c++·算法·排序算法
YuTaoShao6 小时前
【LeetCode 热题 100】206. 反转链表——(解法一)值翻转
算法·leetcode·链表