前缀和——DP34 【模板】前缀和

文章目录

    • [🍋1. 题目](#🍋1. 题目)
    • [🍈2. 算法原理](#🍈2. 算法原理)
    • [🍈3. 代码实现](#🍈3. 代码实现)

🍋1. 题目

题目链接:【模板】前缀和_牛客题霸_牛客网 (nowcoder.com)

描述

给定一个长度为n的数组a1,a2,...an.

接下来有q次查询, 每次查询有两个参数l, r.

对于每个询问, 请输出al + al+1 + ... + ar

输入描述:

第一行包含两个整数n和q.

第二行包含n个整数, 表示a1,a2,...an.

接下来q行,每行包含两个整数 l和r.

1≤n ,q≤105

−109 ≤ a *i* ≤ 109

1≤lrn

输出描述:

输出q行,每行代表一次查询的结果.

示例1

输入:

复制代码
3 2
1 2 4
1 2
2 3

输出:

复制代码
3
6

🍈2. 算法原理

++解法一:暴力模拟++

求哪段区间的和,我们就从指定初始位置一直加到指定结束位置,每次都是在遍历数组,询问q次,每次询问的复杂度为O(n),所以时间复杂度总共就是O(n*q) ,该题数据量十分庞大,nq的范围都是1,10^5^,合起来就是1010,这肯定会超时

++解法二:前缀和++

前缀和思想就是快速求出数组中某一个连续区间的和,一共分为2步:

  1. 预处理出一个前缀和数组dp(原数组和dp数组下标都是从1开始),dp[i]表示[1,i]区间内所以元素的和
  2. 使用前缀和数组
    如果要求区间[l,r],直接拿dp[r] - dp[l-1]即可

这里预处理一个前缀和的数组,需要的时候直接拿值即可,这个时间复杂度为O(1),预处理前缀和数组的复杂度为O(n),所以该方法的复杂度为O(n)+O(q)

细节问题:

这里我们数组的下标不是从0开始,而是从1开始的。

因为这里有个边界问题,如果从0开始,例如求区间[0,3]的结果时,那么我们算出来的就是dp[3]-dp[-1],那么这里就还需要在进行判断一下;如果从1开始,则不需要考虑这个情况,所以默认下标0处的值为0,属于一个辅助节点。

🍈3. 代码实现

cpp 复制代码
#include <iostream>
#include<vector>
using namespace std;

int main()
{
    int n,q;
    cin>>n>>q;
    
    vector<int> arr(n+1);
    for(int i=1;i<=n;i++)   cin>>arr[i];
    
    //预处理前缀和数组
    vector<long long> dp(n+1);  //long long防止溢出
    for(int i=1;i<=n;i++)   dp[i] = dp[i-1]+arr[i];

    int l = 0;
    int r = 0;
    while(q--)
    {
        cin>>l>>r;
        cout<< dp[r]-dp[l-1]<<endl;
    }
    return 0;
}
相关推荐
ShineWinsu11 分钟前
对于Git:基础操作的超详细保姆级解析
linux·c++·git·面试·备份·管理·版本控制器
番茄巴士37 分钟前
手写一个 mini HashMap,彻底搞懂哈希表原理
算法
小小de风呀38 分钟前
de风——【从零开始学C++】(二十)红黑树封装map和set
c++·学习
青梅橘子皮1 小时前
Blue---枚举
c++
宣宣猪的小花园.1 小时前
【机器学习】过拟合与泛化:模型为什么会“刷题很强、实战失灵”
人工智能·算法·机器学习
INGNIGHT1 小时前
624.数组列表中的最大距离(maximum)
算法·散列表
Niuguangshuo1 小时前
论文解读:w2v-BERT,把 wav2vec 2.0 和 BERT 合成一根管子的语音 SSL
算法·音视频·语音识别
科技小E1 小时前
国标视频分析平台EasyGBS×自动化AI算法训练服务器DLTM,把通用AI炼成你的现场AI
算法·自动化·音视频
zander2581 小时前
LeetCode 15. 三数之和
算法
AiNightVision1 小时前
NMC存算一体与AI ISP
人工智能·算法·车载系统·自动驾驶·无人机·视频·智能硬件