C++如何实现高效的数组前缀和计算(一维与多维)

冬宇同学_9982

冬宇同学_9982

2026-06-12

696人浏览

原创

一维前缀和推荐用std::partial_sum,安全处理边界;二维需手动先每行再每列计算,注意索引偏移和缓存友好性。

c++如何实现高效的数组前缀和计算(一维与多维)

一维数组前缀和:用 std::partial_sum 比手写 for 循环更安全

直接调 std::partial_sum 是最稳妥的选择,它默认做累加(std::plus),且能正确处理空数组、单元素、迭代器失效等边界。手写循环容易漏掉 prefix[0] = arr[0] 或越界访问。

  • 输入必须是可随机访问的容器(std::vector、原生数组),不能是 std::list
  • 输出容器需预先分配空间(或用 back_inserter),否则会写入未初始化内存
  • 若需原地计算(复用原数组存前缀和),确保输入输出迭代器不重叠;否则用 std::vector 临时拷贝再覆盖

示例:

std::vector<int> arr = {1, 2, 3, 4};<br>std::vector<int> prefix(arr.size());<br>std::partial_sum(arr.begin(), arr.end(), prefix.begin()); // → {1, 3, 6, 10}</int></int>

二维数组前缀和:别用嵌套 std::partial_sum,先按行再按列

二维前缀和不是简单套用两次 std::partial_sum 就行。标准库没有现成的二维版本,必须手动实现:先对每行做一维前缀和,再对结果矩阵的每一列做一维前缀和。顺序不能反,否则逻辑错。

  • 行优先遍历保证缓存友好;列优先会导致大量 cache miss
  • 注意索引偏移:二维前缀和通常定义为 sum[i][j] 表示从 (0,0) 到 (i,j) 的子矩阵和,因此需预留第 0 行/列全 0,或统一用 1-based 索引
  • 若原矩阵是 std::vector<:vector>></:vector>,确保每行长度一致,否则运行时崩溃

关键步骤:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
// 假设 mat 是 h×w 的 vector<vector>>,res 初始化为 (h+1)×(w+1)<br>for (int i = 1; i   for (int j = 1; j     res[i][j] = mat[i-1][j-1] + res[i-1][j] + res[i][j-1] - res[i-1][j-1];<br>  }<br>}</vector>

性能陷阱:前缀和构建后查询快,但修改代价高

前缀和结构天生不支持高效单点更新。一旦数组某元素变更,一维需 O(n) 重建,二维需 O(h×w)。这不是算法缺陷,而是设计取舍——它只适合“构建一次、查询多次、几乎不修改”的场景。

  • 若存在频繁更新,改用 std::vector + std::accumulate 每次查区间和(O(n) 查询);或换 fenwick tree(树状数组)
  • 多维场景下,std::vector<:vector>></:vector> 的连续性差于一维扁平化存储;用 std::vector<int> flat(h * w)</int> 并手动映射坐标,能提升构建与查询局部性
  • 编译器对 std::partial_sum 有优化,但对自定义二维循环未必自动向量化;如需极致性能,考虑 #pragma omp simd 或手动展开内层循环

边界与类型:小心整数溢出和 signed/unsigned 混用

前缀和值域可能远超原数组元素类型范围。比如 vector<uint8_t></uint8_t> 有 255 个元素,前缀和最大可达 255×255=65025,超出 uint8_t 上限。

  • 构建前缀和时,务必用足够宽的有符号类型(如 long long)承载累加结果,尤其在竞赛或大数据量场景
  • 避免 unsigned 类型参与减法运算(如二维前缀和公式中的 - res[i-1][j-1]),否则可能因回绕导致巨大正数
  • STL 算法如 std::partial_sum 不检查溢出,也不会抛异常,结果静默错误

实际编码中,宁可显式声明:

std::vector<long long> prefix(arr.size());<br>std::partial_sum(arr.begin(), arr.end(), prefix.begin());</long>

构建二维前缀和时最容易忽略的是内存布局与索引偏移的耦合——同一份数据,用扁平化一维数组存还是用 vector of vector 存,直接影响 cache 命中率和边界判断方式。这点在真实项目里调试起来非常隐蔽。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

c++

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

2248

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

1019

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

447

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

327

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

426

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

580

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1409

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

1217

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1325

9

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习