0

0

Python怎么求前缀和_一维与二维前缀和数组快速区间查询

P粉602998670

P粉602998670

发布时间:2026-03-19 13:11:03

|

666人浏览过

|

来源于php中文网

原创

一维前缀和用长度n+1数组存储,prefix[i]表示前i个元素和,查询[l,r]为prefix[r+1]-prefix[l];二维需多开行列,用容斥公式prefixr2+1-prefixr1-prefixr2+1+prefixr1。

python怎么求前缀和_一维与二维前缀和数组快速区间查询

一维前缀和:用 list 累加构造,查询靠下标减法

一维前缀和本质是把「从开头到每个位置的累加值」存下来,后续查任意区间 [l, r] 的和就不用再循环加一遍。关键不是“怎么算”,而是“下标对不对”——多数人栽在边界上。

  • prefix[i] 通常定义为 nums[0] + nums[1] + ... + nums[i-1](即 prefix[0] = 0),这样 query(l, r) 就是 prefix[r+1] - prefix[l],不用特判 l == 0
  • 别用 prefix[i] = sum(nums[:i]) 实时切片求和——时间退化成 O(n²),老老实实用一次遍历:
    prefix = [0] * (n + 1)<br>for i in range(n):<br>  prefix[i + 1] = prefix[i] + nums[i]
  • 如果原数组下标从 1 开始(比如题目给的是 1-indexed 输入),别硬套 0-indexed 模板,先统一转成 0-indexed 再建前缀和,否则 l/r 映射错一位,结果全偏

二维前缀和:按行优先展开成二维 dp,容斥原理绕不开

二维前缀和不是“每行单独做一维”,而是把左上角 (0,0) 到当前点 (i,j) 的矩形和存下来。核心公式是:prefix[i+1][j+1] = prefix[i][j+1] + prefix[i+1][j] - prefix[i][j] + matrix[i][j]。漏掉中间的减法,整个数组就溢出。

  • 必须用 (i+1, j+1) 开辟多一行一列,让 prefix[0][*]prefix[*][0] 全为 0,否则查 (0,0)(r,c) 时要写一堆 if 判断
  • 查子矩阵 [r1, c1][r2, c2](闭区间)的和,公式固定为:prefix[r2+1][c2+1] - prefix[r1][c2+1] - prefix[r2+1][c1] + prefix[r1][c1]。记混加减顺序,结果必错;建议画个图,把四个角对应区域标出来再推
  • 初始化二维 prefix 时别用 [[0] * (m+1)] * (n+1)——这是浅拷贝,改一行全变。老实用嵌套列表推导:[[0] * (m + 1) for _ in range(n + 1)]

修改频繁?前缀和就不该上场

前缀和只适合「静态数组 + 大量查询」场景。一旦有单点修改(比如 nums[i] += x),重算整个前缀和是 O(n) 或 O(nm),比暴力还慢。

  • 需要边改边查,直接换 fenwick tree(树状数组)或 segment tree(线段树)——它们单次修改+查询都是 O(log n)
  • 真想硬用前缀和顶着改?那每次修改后必须调用完整重建函数,别只更新一个位置,否则后续所有查询都错
  • LeetCode 上标“前缀和”的题,90% 不带修改;但看到“数据流”“在线查询”“update 方法”等字眼,立刻放弃前缀和思路

Python 实现注意:别用 numpy.cumsum 替代手写

numpy.cumsum 虽快,但它返回的是新数组,且默认展平多维数组。用它做二维前缀和,要么手动 reshape 再逐行处理,要么结果维度错乱,反而更难 debug。

Boba.video
Boba.video

AI动漫视频生成器

下载

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

  • 纯 Python 场景下,手写循环比依赖 numpy 更可控、更符合算法题约束(很多 OJ 不开 numpy)
  • 如果坚持用 numpy,二维必须指定 axisnp.cumsum(matrix, axis=0) 是列方向累加,axis=1 是行方向——但这只是“行/列前缀和”,不是真正的二维矩形前缀和
  • 真正二维前缀和在 numpy 里没内置函数,硬写也得按容斥公式来,不如回归原生逻辑

实际写的时候,最常漏的是二维容斥里的那个加回项,或者一维 prefix 长度多开/少开 1。这些地方不画小例子跑两组数,光看公式根本记不住。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

781

2023.06.15

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

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

697

2023.07.20

python能做什么
python能做什么

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

784

2023.07.25

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

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

890

2023.07.31

python教程
python教程

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

1564

2023.08.03

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

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

598

2023.08.04

python eval
python eval

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

591

2023.08.04

scratch和python区别
scratch和python区别

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

759

2023.08.11

Go Web框架Gin接口开发与中间件设计实践
Go Web框架Gin接口开发与中间件设计实践

本专题围绕 Go 在 Web 后端开发中的主流框架 Gin 展开,系统讲解高性能接口开发与中间件机制设计。内容涵盖路由分组、请求绑定、参数校验、统一响应封装、日志与鉴权中间件实现,以及接口限流与异常处理策略。通过实战项目案例,帮助开发者构建结构清晰、性能优良的 Go Web 服务体系,提升接口开发效率与系统可维护性。

7

2026.03.19

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 2万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号