蓝桥杯-统计子矩阵

本文介绍了一种解决二维区间前缀和计算问题的方法,通过使用二维数组和前缀和的思路,实现对给定矩阵中满足特定条件的子矩阵数量的计数,目标是使子矩阵的前缀和不超过给定值k。
"""
https://www.lanqiao.cn/problems/2109/learning/?page=1&first_category_id=1&name=%E7%BB%9F%E8%AE%A1%E5%AD%90%E7%9F%A9%E9%98%B5
"""
# 二维前缀和
# 利用前缀和思路只能通过70%
n, m, k = map(int, input().split())
# 下标从1开始
a = [[0] * (m + 1) for i in range(n + 1)]
sum = [[0] * (m + 1) for i in range(n + 1)]

# 输入一个二维数组
for i in range(1, n + 1):
  a[i] = [0] + list(map(int, input().split()))

# 求二维前缀和
for i in range(1, n + 1):
  for j in range(1, m + 1):
    sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + a[i][j]

# 求二维区间前缀和
def get_sum(sum, x1, y1, x2, y2):
  return sum[x2][y2] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1]

ans = 0
# 枚举矩阵左上角
for x1 in range(1, n + 1):
  for y1 in range(1, m + 1):
    # 枚举矩阵右下角
    for x2 in range(x1, n + 1):
      for y2 in range(y1, m + 1):
        if get_sum(sum, x1, y1, x2, y2) <= k:
          ans += 1
print(ans)

 

 

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

25年上工地

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值