【算法】一维前缀和

趴抖

已于 2022-12-22 22:50:56 修改

阅读量122

点赞数 4

文章标签：学习算法 c++

于 2022-11-23 20:05:20 首次发布

本文链接：https://blog.csdn.net/PuddleRubbish/article/details/127971926

版权

前缀和区间求和算法优化时间复杂度动态规划

关键词由CSDN通过智能技术生成

声明：本文仅为学习笔记，不作他用

前缀和

一维前缀和

输入一个长度为n的整数序列。接下来再输入m个询问，每个询问输入一对l, r。对于每个询问，输出原序列中从第l个数到第r个数的和。

第一种方法：遍历区间求和

代码如下：

const int N = 1e5 + 10;
int a[N];
int n,m;
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
while(m--)
{
    int l, r;
    int sum = 0;
    scanf("%d%d", &l, &r);
    for(int i = l; i <= r; i++)
    { 
        sum += a[i];
    }
    printf("%d\n",sum);
}

这样的时间复杂度为O(n * m)，如果n和m的数据量稍微大一点就有可能超时，而我们如果使用前缀和的方法来做的话就能够将时间复杂度降到O(n + m)，大大提高了运算效率。

故第二种方法：前缀和

原理如下：
sum[r] = a[1] + a[2] + a[3] + a[l-1] + a[l] + a[l + 1] … a[r];
sum[l - 1] = a[1] + a[2] + a[3] + a[l - 1];
则对于询问[l,r]，有：
sum[r] - sum[l - 1] = a[l] + a[l + 1] + …+ a[r];
在这里插入图片描述这样，对于每个询问，只需要执行 sum[r] - sum[l - 1]。输出原序列中从第l个数到第r个数的和的时间复杂度变成了O(1)。

我们把它叫做一维前缀和。

#include <cstdio>
#include <iostream>
using namespace std;
const int N = 1e7+5;
int n,Q,a[N],sum[N];
int main(){
	cin>>n;
	for(int i = 1; i <= n; ++i){
		cin>>a[i];
		sum[i]=sum[i-1]+a[i];
	}
	cin>>Q;
	while(Q--){
		int L,R;
		cin>>L>>R;
		printf("%d\n",sum[R]-sum[L-1]);
	}
}

例.静态区间求和

仅询问而不修改。
给定一个数列A，多次询问区间和。
对于数列1,5,3,2,4询问[2,5]时输出14

#include<stdio.h>
int main() {
    int l, r;
    int n;
    scanf_s("%d", &n);
    int num;
    const int N = 1e5 + 10;
    int sum[N], a[N]; //sum[i]=a[1]+a[2]+a[3].....a[i];
    sum[0] = 0;
    for (int i = 1; i <= n; i++)
    {
        scanf_s("%d", &num);
        a[i] = num;
        sum[i] = sum[i - 1] + a[i];
    }
    scanf_s("%d%d", &l, &r);
    printf("%d\n", sum[r] - sum[l - 1]);
}