子矩阵(c++实现)

题目


给定一个 n×m (n 行 m 列)的矩阵。

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

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

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


输入


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

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


输出


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


样例


输入样例:

2 3 1 2

1 2 3

4 5 6


输出样例:

58


CODE

c++ 复制代码
#include<iostream>
using namespace std;
const int N = 1010,MOD = 998244353;
int w[N][N];
int n,m,A,B;
int rmax[N][N],rmin[N][N];
int q[N];

void get_max(int a[],int b[],int tot,int k){
    int hh=0,tt=-1;
    for(int i=0;i<tot;i++){
        if(hh<=tt && i-k>=q[hh]) 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 tot,int k){
    int hh=0,tt=-1;
    for(int i=0;i<tot;i++){
        if(hh<=tt && i-k>=q[hh]) hh++;
        while(hh<=tt && a[q[tt]]>=a[i]) tt--;
        q[++tt] = i;
        b[i] = a[q[hh]];
    }
}

int main(){
    scanf("%d%d%d%d",&n,&m,&A,&B);
    
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            scanf("%d",&w[i][j]);
        }
    }
    
    for(int i=0;i<n;i++){
        get_max(w[i],rmax[i],m,B);
        get_min(w[i],rmin[i],m,B);
    }
    
    int res = 0;
    int a[N],b[N],c[N];
    
    for(int i=B-1;i<m;i++){
        for(int j=0;j<n;j++) a[j] = rmax[j][i];
        get_max(a,b,n,A);
        for(int j=0;j<n;j++) a[j] = rmin[j][i];
        get_min(a,c,n,A);
        for(int j=A-1;j<n;j++){
            res = (res + (long long)b[j]*c[j])%MOD;
        }
    }
    printf("%d",res);
}
相关推荐
今天背单词了吗9804 分钟前
算法学习笔记:8.Bellman-Ford 算法——从原理到实战,涵盖 LeetCode 与考研 408 例题
java·开发语言·后端·算法·最短路径问题
手握风云-14 分钟前
优选算法的链脉之韵:链表专题
数据结构·算法·链表
Coding小公仔18 分钟前
LeetCode 151. 反转字符串中的单词
开发语言·c++·算法
稳兽龙18 分钟前
P1098 [NOIP 2007 提高组] 字符串的展开
c++·算法·模拟
G.E.N.23 分钟前
开源!RAG竞技场(2):标准RAG算法
大数据·人工智能·深度学习·神经网络·算法·llm·rag
写个博客27 分钟前
暑假算法日记第三天
算法
✿ ༺ ོIT技术༻1 小时前
剑指offer第2版:动态规划+记忆化搜索
算法·动态规划·记忆化搜索
算法练习生1 小时前
Qt核心类QWidget及其派生类详解
开发语言·c++·qt
oioihoii1 小时前
C++11标准库算法:深入理解std::none_of
java·c++·算法
老虎06272 小时前
数据结构(Java)--位运算
java·开发语言·数据结构