【算法 C/C++】一维前缀和

2025 - 03 - 08 - 第 68 篇
Author: 郑龙浩 / 仟濹
【一维前缀和】

文章目录

  • [前缀和与差分 - 我的博客](#前缀和与差分 - 我的博客)
    • [1 大体介绍](#1 大体介绍)
    • [2 计算某些区间的和( 不使用前缀和 )](#2 计算某些区间的和( 不使用前缀和 ))
    • [3 计算某些区间的和( 使用前缀和 )](#3 计算某些区间的和( 使用前缀和 ))

前缀和与差分 - 我的博客

一维前缀和
【算法 C/C++】一维前缀和
一维差分
【算法 C/C++】一维差分
二维前缀和
【算法 C/C++】二维前缀和
二维差分
【算法 C/C++】二维差分# 前缀和(一维)

时间复杂度为 O(1)

可以在O(1)的时间复杂度内获取任意子数组的和。

1 大体介绍

是一种用于计算区间和的问题的高效算法,尤其适用于静态数组(即数组在查询时是不会改变的)。它通过预先计算前缀和数组来快速响应区间和查询,时间复杂度为 O(1),空间复杂度为 O(n),适用于多次查询的场景。

2 计算某些区间的和( 不使用前缀和 )

题目

有一个数组 arr[5] = {1, 3, 7, 5, 2},求如下区间的和

[3, 4] [2, 4] [0, 3]??

答案为:7 14 16

思路

如果不使用前缀和的话,会非常的麻烦,且如果求得区间非常多,且数值较大,时间复杂度会非常大。下面写的是不使用前缀和的时候的方法,纯循环。

代码

cpp 复制代码
#include <algorithm>
#include <iostream>
using namespace std;
int arr[5] = {1, 3, 7, 5, 2};
int sum(int a, int b){
    int ans = 0;
    for( int i = a; i <= b; i++ ){
        ans += arr[ i ];
    }
    return ans;
}
int main( void ){
    cout << sum(3, 4) << " " << sum(2, 4) << " " << sum(0, 3) << endl;
    return 0;
}

3 计算某些区间的和( 使用前缀和 )

题目

有一个数组 arr[5] = {1, 3, 7, 5, 2},求如下区间的和

[3, 4] [2, 4] [0, 3]??

答案为:7 14 16

思路

提前写一个数组比如sum,在 sum[i] 中存储 sum[0]到sum[i] 的和。

前缀和的方法:当求某个区间比如 [2, 4] 的时候,只需要计算 [0, 4] 的和 - [0, 1] 的和,也就是 sum[4] - sum[1]

即,如果求[a, b]的和,只需要 sum[b] - sum[a - 1]

这样的好处是,当需要计算多个区间的时候,节省了计算时间,比如[a, b],之前的方式是从 arr[a]循环累加到arr[b],学习了前缀和以后无需这么麻烦,只需要提前将 arr[i]arr[0]的和计算出来,当需要知道某个区间的和的时候,只需要进行不同位置的和的相减计算即可。

代码1 - 使用函数

cpp 复制代码
// 使用前缀和计算区间
// 有一个数组 `arr[5] = {1, 3, 7, 5, 2}`,求如下区间的和
// `[3, 4] [2, 4] [0, 3]`??
// 答案为:`7 14 16`
#include <algorithm>
#include <iostream>
using namespace std;
#define N 5
int arr[5] = {1, 3, 7, 5, 2};
int sum[5];
int get_sum(int a, int b){
    // 如果计算的是 [0, b],直接打印 sum[b] 即可 --> sum[b] 就是 [0, b] 的和
    if( a == 0 ){
        return sum[ b ];
    } else {
        return sum[ b ] - sum[ a - 1 ];
    }
}
int main( void ){
    sum[ 0 ] = arr[ 0 ]; // 第一个元素到第一个元素的和为第一个元素
    for( int i = 1; i < N; i++ ){
        sum[ i ] = sum[i - 1] + arr[ i ];
    }
    cout << get_sum(3, 4) << " " << get_sum(2, 4) << " " << get_sum(0, 3) << endl;
    return 0;
}

代码2 - 使用字符串替换 #define

更简洁高效的代码

cpp 复制代码
// 使用前缀和计算区间
// 有一个数组 `arr[5] = {1, 3, 7, 5, 2}`,求如下区间的和
// `[3, 4] [2, 4] [0, 3]`??
// 答案为:`7 14 16`
#include <algorithm>
#include <iostream>
using namespace std;
#define get_sum(a, b) (a ? sum[b] - sum[a - 1] : sum[b])
#define N 5
int arr[5] = {1, 3, 7, 5, 2};
int sum[5];
int main( void ){
    sum[ 0 ] = arr[ 0 ]; // 第一个元素到第一个元素的和为第一个元素
    for( int i = 1; i < N; i++ ){
        sum[ i ] = sum[i - 1] + arr[ i ];
    }
    cout << get_sum(3, 4) << " " << get_sum(2, 4) << " " << get_sum(0, 3) << endl;
    return 0;
}
相关推荐
向阳@向远方1 小时前
第二章 简单程序设计
开发语言·c++·算法
Mr_Xuhhh1 小时前
信号与槽的总结
java·开发语言·数据库·c++·qt·系统架构
github_czy2 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
liulilittle2 小时前
VGW 虚拟网关用户手册 (PPP PRIVATE NETWORK 基础设施)
开发语言·网络·c++·网关·智能路由器·路由器·通信
许愿与你永世安宁2 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子2 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
ruanjiananquan992 小时前
c,c++语言的栈内存、堆内存及任意读写内存
java·c语言·c++
满分观察网友z3 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
持梦远方3 小时前
C 语言基础入门:基本数据类型与运算符详解
c语言·开发语言·c++
YuTaoShao3 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵