统计子矩阵

问题描述

给定一个NXM的矩A请你统计有多少个子矩阵(最小1x1最大M)满足子矩阵中所有数的和不超过给定的数K?

输入格式

第一行包含三个整数N,M和K

之后N行每行包含M个整数代表矩阵A

输出格式

个整数代表答案

样例输入

3 4 10

1 2 34

5 6 7 8

9 10 11 12

样例输出

19

复制代码
import os
import sys

# 请在此输入您的代码
n,m,k=map(int,input().split())
s=[[0 for i in range(m+1)]]  #构建二维列表

for i in range(1,n+1):
  s.append([0]+list(map(int,input().split())))  #[0,1,2,3,4]
  for j in range(1,m+1):
    s[i][j]+=s[i-1][j]  #求每列的累加和更新到每列上去

ans=0
for i in range(1,n+1):#确定上边界 ,如i=1,第一行
  for j in range(i,n+1): #确定下边界,j=2,3,4...,第三行,可求这三行中的矩阵
    l=1 #左边界
    sum=0
    for r in range(1,m+1): #遍历每一列
      sum+=s[j][r]-s[i-1][r] #i-1是因为第一行全是0,也避免i,j都等1的情况,因为s已经求累加,二者相减求得矩阵中所有数和
      while sum>k:
        sum-=s[j][l]-s[i-1][l]    #减去左边得和
        l+=1  #超过k的话更新左边界,缩小矩阵左边大小,使矩阵继续向右边扩大
      ans+=r-l+1 #从1到3,3-1+1=3

print(ans)

由于时间限制问题,只能通过70%

相关推荐
WWZZ202530 分钟前
ORB_SLAM2原理及代码解析:SetPose() 函数
人工智能·opencv·算法·计算机视觉·机器人·自动驾驶
小马学嵌入式~1 小时前
堆排序原理与实现详解
开发语言·数据结构·学习·算法
青岛少儿编程-王老师1 小时前
CCF编程能力等级认证GESP—C++6级—20250927
java·c++·算法
一人の梅雨1 小时前
1688 拍立淘接口深度开发:从图像识别到供应链匹配的技术实现
人工智能·算法·计算机视觉
Miraitowa_cheems2 小时前
LeetCode算法日记 - Day 64: 岛屿的最大面积、被围绕的区域
java·算法·leetcode·决策树·职场和发展·深度优先·推荐算法
Christo32 小时前
关于K-means和FCM的凸性问题讨论
人工智能·算法·机器学习·数据挖掘·kmeans
_不会dp不改名_3 小时前
leetcode_1382 将二叉搜索树变平衡树
算法·leetcode·职场和发展
greentea_20133 小时前
Codeforces Round 173 B. Digits(2043)
c++·算法
m0_743106464 小时前
LOBE-GS:分块&致密化效率提升
人工智能·算法·计算机视觉·3d·几何学
徐子童4 小时前
优选算法---字符串
java·算法·字符串·笔试·高精度相乘