关于二维/多维前缀和

二维/多维前缀和

将一维前缀和拓展到多维的情形,就是多维前缀和.常见的多维前缀和的求解方法有两种.

基于容斥原理

这种方法多用于二维前缀和的情形.给定大小为 𝑚 ×𝑛 的二维数组 𝐴,要求出其前缀和 𝑆.那么,𝑆 同样是大小为 𝑚 ×𝑛 的二维数组,且

𝑆𝑖,𝑗=∑𝑖′≤𝑖∑𝑗′≤𝑗𝐴𝑖′,𝑗′.

类比一维的情形,𝑆𝑖,𝑗 应该可以基于 𝑆𝑖−1,𝑗 或 𝑆𝑖,𝑗−1 计算,从而避免重复计算前面若干项的和.但是,如果直接将 𝑆𝑖−1,𝑗 和 𝑆𝑖,𝑗−1 相加,再加上 𝐴𝑖,𝑗,会导致重复计算 𝑆𝑖−1,𝑗−1 这一重叠部分的前缀和,所以还需要再将这部分减掉.这就是容斥原理.由此得到如下递推关系:

𝑆𝑖,𝑗=𝐴𝑖,𝑗+𝑆𝑖−1,𝑗+𝑆𝑖,𝑗−1−𝑆𝑖−1,𝑗−1.

实现时,直接遍历 (𝑖,𝑗) 求和即可.

同样的道理,在已经预处理出二维前缀和后,要查询左上角为 (𝑖1,𝑗1)、右下角为 (𝑖2,𝑗2) 的子矩阵的和,可以计算

𝑆𝑖2,𝑗2−𝑆𝑖1−1,𝑗2−𝑆𝑖2,𝑗1−1+𝑆𝑖1−1,𝑗1−1.

这可以在 𝑂(1) 时间内完成.

在二维的情形,以上算法的时间复杂度可以简单认为是 𝑂(𝑚𝑛),即与给定数组的大小成线性关系.但是,当维度 𝑘 增大时,由于容斥原理涉及的项数以指数级的速度增长,时间复杂度会成为 𝑂(2𝑘𝑁),其中 𝑘 是数组维度,而 𝑁 是给定数组大小.因此,该算法不再适用.

给定数组大小.因此,该算法不再适用.

详见前缀和 & 差分 - OI Wiki

相关推荐
GreenTea12 分钟前
🔥别再用压缩了,Codex、NVIDIA、DeepSeek、Uber 给出 Agent 上下文管理的新答案
前端·后端·算法
All for pursuit.24 分钟前
【矩阵-2】240.搜索二维矩阵 II
数据结构·c++·算法·leetcode
6Hzlia40 分钟前
【Classic 150 刷题计划】 LeetCode 209. 长度最小的子数组 | C++ 滑动窗口(毛毛虫算法)经典模板
c++·算法·leetcode
6Hzlia1 小时前
【Classic 150 刷题计划】 LeetCode 28. 找出字符串中第一个匹配项的下标 | C++ 滑动窗口与子串比对
c++·算法·leetcode
GreenTea9 小时前
vLLM 与 SGLang KV Cache 底层实现机制深度调研报告
前端·后端·算法
Bode_20029 小时前
多资源规划(含生产调度、库存管理及产能规划)的创新优化算法
算法·调度
土司大王10 小时前
LeetCode hot100——35.搜索插入位置:Java 二分模板、左闭右开区间与插入点分析
java·算法·leetcode
微三云生态系统架构师-彭丹10 小时前
远方好物S2B2C系统架构:一级分销与保证金托管的合规技术实现
人工智能·算法
土司大王11 小时前
LeetCode hot100——34.在排序数组中查找元素的第一个和最后一个位置:Java 二分模板、边界分析
java·算法·leetcode
罗西的思考12 小时前
机器人模型(WM / WAM / VLA)综合分析与对比:从「看」到「想」再到「做」
人工智能·算法·机器学习