讲师中心 微信公众号

工具推荐

视频效率加速
剧云
剧云

中文短剧剧本制作平台

AniShort
AniShort

AI短剧协同制作平台

updream

一键生成高质量推广成片

Seko
Seko

AI短剧一键生成成片

如何在Python中利用前缀和数组快速求多维区间总和

胖明大大_4071

胖明大大_4071

发布时间:2026-11-10 08:50:30

|

907人浏览过

|

来源于php中文网

原创

二维前缀和数组需扩容一行一列并用0填充,原数组元素从(1,1)开始映射;查询子矩阵和时,0-based坐标需各加1再代入公式dpr2+1-dpr1-dpr2+1+dpr1。

如何在python中利用前缀和数组快速求多维区间总和

二维前缀和数组怎么构建才不会越界

构建二维前缀和数组时,最常踩的坑是索引偏移没处理好,导致 dp[i][j] 计算时访问了不存在的 dp[i-1][j] 或 dp[i][j-1]。正确做法是让前缀和数组比原数组多一行一列,全部用 0 填充,原数组 matrix 的元素从 (1,1) 开始映射。

核心递推式是:dp[i][j] = matrix[i-1][j-1] + dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1]。这个公式本质是容斥:把左上矩形、上方矩形、左方矩形加起来,再减掉重复计算的左上角重叠部分。

常见错误现象:IndexError: list index out of range,往往因为忘了扩容或错把 i,j 当作原数组下标直接代入。

如何用二维前缀和查任意子矩阵和(含边界细节)

给定子矩阵左上角 (r1, c1)、右下角 (r2, c2)(均为 0-based),对应前缀和数组 dp(1-based 索引),查询公式为:dp[r2+1][c2+1] - dp[r1][c2+1] - dp[r2+1][c1] + dp[r1][c1]。

立即学习“Python免费学习笔记(深入)”;

关键点在于:前缀和数组的 (i,j) 存的是从 (0,0) 到 (i-1,j-1) 的和,所以原坐标必须 +1 才能对齐。漏加这 1 就会少算一行或一列。

使用场景举例:图像像素块求和、网格中频繁查询矩形区域数值总和(如游戏地图资源统计、热力图聚合)。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • 若输入坐标是 1-based,先统一转成 0-based 再套公式
  • 务必检查 r1 且 <code>c1 ,否则结果无意义
  • 所有下标在代入 dp 前必须保证 ≥ 0 且 ≤ len(dp)

三维前缀和还能用吗?空间和时间代价是否合理

可以,但要谨慎。三维前缀和构建公式是:dp[i][j][k] = arr[i-1][j-1][k-1] + dp[i-1][j][k] + dp[i][j-1][k] + dp[i][j][k-1] - dp[i-1][j-1][k] - dp[i-1][j][k-1] - dp[i][j-1][k-1] + dp[i-1][j-1][k-1],共 8 项,是标准容斥展开。

问题在于空间爆炸:若原始数组是 100×100×100,dp 就要占用约 100³ × 8 字节 ≈ 80 MB(Python 中每个 int 更大,实际可能翻倍)。查询一次虽仍是 O(1),但预处理时间 O(n³) 和内存占用常成为瓶颈。

替代思路:如果查询不频繁,或只沿某一维切片查询,可考虑二维前缀和嵌套 + 一维遍历,而非硬上三维。

NumPy 里有没有现成的高效替代方案

有,但不是直接叫“prefix_sum”。最实用的是 np.cumsum 配合轴向操作,再结合切片差分。

例如二维情形:

import numpy as np
arr = np.array([[1,2,3],[4,5,6],[7,8,9]])
dp = np.pad(np.cumsum(np.cumsum(arr, axis=0), axis=1), ((1,0),(1,0)), 'constant')
# 然后按前述公式用 dp 查询,注意 np.pad 默认补 0

优势是底层 C 实现,构建比纯 Python 快数倍;劣势是仍需手动处理索引偏移,且三维时 np.cumsum 轴向链式调用易出错。更稳妥的做法是封装一个函数,把坐标转换和差分逻辑收进去,避免每次手算 +1/-1。

真正容易被忽略的是:NumPy 数组默认是 C-order,cumsum 沿 axis=0 是列优先累加,和数学习惯一致;但如果原始数据是 Fortran-order 或做了 transpose,结果会错位。

热门AI工具

更多
切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1891

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

5264

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1849

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

31797

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3767

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3767

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1263

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

40

2026.11.06

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn