单调队列(二维)

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。

思路 (将二维窗口转变为一维窗口)

我们假定有这么一个n*m矩阵(蓝色部分),它有一个A * B 的子矩阵(紫色部分)
在这里插入图片描述
我们可以先求出子矩阵每一行的最大值,再从这些最大值中找出最大的元素,就求出这一个子矩阵的最大值了。

具体操作:
先求 蓝色矩阵 每行滑动窗口的最大值,就可以得到每一个 紫色矩阵 每一行的最大值;
同理求最小值。

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

using namespace std;

typedef long long ll;
const int N=1010;
int g[N][N],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+1>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+1>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",&g[i][j]);
    
    for(int i=0;i<n;i++)
    {
        get_max(g[i],rmax[i],m,B);
        get_min(g[i],rmin[i],m,B);
    }
    
    int a[N],b[N],c[N],res=0;
    
    for(int j=B-1 ;j<m; j++)
    {
        for(int i=0;i<n;i++) a[i]=rmax[i][j];
        
        get_max(a,b,n,A);
        
        for(int i=0;i<n;i++) a[i]=rmin[i][j];
        
        get_min(a,c,n,A);
        
        for(int i=A-1;i<n;i++) res=(res+(ll)b[i]*c[i])%998244353;
    }
    
    cout<<res<<endl;
    
    return 0;
}
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值