蓝桥杯真题 2109.统计子矩阵

原题地址:1.统计子矩阵 - 蓝桥云课

问题描述

给定一个 N×MN×M 的矩阵 AA, 请你统计有多少个子矩阵 (最小 1×11×1, 最大 N×M)N×M) 满足子矩阵中所有数的和不超过给定的整数 KK ?

输入格式

第一行包含三个整数 N,MN,M 和 KK.

之后 NN 行每行包含 MM 个整数, 代表矩阵 AA.

输出格式

一个整数代表答案。

具体代码和思路如下:

cpp 复制代码
#include <iostream>
using namespace std;

int n,m,k;
int a[501][501];
long long ans = 0;

int main()
{
    cin>>n>>m>>k;
    for (int i = 1;i <= n;++i){
        for (int j = 1;j <= m;++j){
            cin>>a[i][j];
            //将每一行都看成一个整数,来进行前缀和的操作
            a[i][j] += a[i - 1][j];
        }
    }
    for (int i = 1;i <= n;++i){//滑动窗口的上边界
        for (int ii = i;ii <= n;++ii){//滑动窗口的下边界
            int l = 1,r = 1;//l为滑动窗口的左边界,r为滑动窗口的右边界
        int sum = 0;//sum为当前滑动窗口内所有元素的和,即小区间[l,r]的和
            for (r = 1;r <= m;++r){//更新右边界
                sum += a[ii][r] - a[i - 1][r];//加入新扩大的窗口的边界值
                while (sum > k){//当sum大于k时,就停止更新右边界r,开始让左边界l向右移,直到区间[l,r]内的值小于等于k
                    sum -= a[ii][l] - a[i - 1][l];//由于左边界l要向右移,所以要减去原来左边界上的值
                    l++;
                }
                //当区间[l,r]内的所有元素和小于等于k时,就说明该区间内的所有子区间的区间和都要小于等于k
                ans += r - l + 1;
            }
        }
    }
    cout<<ans;
    return 0;
}
相关推荐
躲着人群26 分钟前
次短路&&P2865 [USACO06NOV] Roadblocks G题解
c语言·数据结构·c++·算法·dijkstra·次短路
心动啊1211 小时前
支持向量机
算法·机器学习·支持向量机
小欣加油2 小时前
leetcode 1493 删掉一个元素以后全为1的最长子数组
c++·算法·leetcode
蓝风破云3 小时前
C++实现常见的排序算法
数据结构·c++·算法·排序算法·visual studio
艾醒3 小时前
大模型面试题剖析:Pre-Norm与Post-Norm的对比及当代大模型选择Pre-Norm的原因
算法
怀旧,3 小时前
【C++】 9. vector
java·c++·算法
浩浩测试一下4 小时前
06高级语言逻辑结构到汇编语言之逻辑结构转换 for (...; ...; ...)
汇编·数据结构·算法·安全·web安全·网络安全·安全架构
辞--忧5 小时前
K-Means 聚类算法详解与实战指南
算法·kmeans·聚类
尤超宇5 小时前
K 均值聚类(K-Means)演示,通过生成笑脸和爱心两种形状的模拟数据,展示了无监督学习中聚类算法的效果。以下是详细讲解:
算法·均值算法·聚类
qq_479875436 小时前
设置接收超时(SO_RCVTIMEO)
c语言·算法