一维前缀和

        前缀和是指某序列的前n项和,可以把它理解为数学上的数列的前n项和,而差分可以看成前缀和的逆运算。合理的使用前缀和与差分,可以将某些复杂的问题简单化。

一、一维前缀和

       输入一个长度为n的序列,对这个序列进行m次询问,每次询问输入两个整数l和r,求第l个到第r个区间序列的和。

数据范围:1 <= n,m <= 10^5;l,r <= n。

       我们很容易想到暴力的解法,遍历区间。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int arr[N];
int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++) cin >> arr[i];
    while(m--){
        int l, r;
        cin >> l >> r;
        int ans = 0;
        for(int i = l; i <= r; i++){
            ans += arr[i];
        }
        cout << ans << endl;
    }
    return 0;
}

        不难看出上面代码的时间复杂度最坏情况为O(n^2),如果n = 10^5的话,这毫无疑问会超时,所以我们要利用前缀和的方法进行解题。

        接下来我们说一下前缀和。

        首先我们给出一个序列:arr[7] = {1,5, 2, 7, 9, 3};

     那么它的前缀和序列就为:sum[7] = {1,6,8,15,24,27};

        我们假设l = 2, r = 4;不难从元序列看出和为:5 + 2 + 7 = 14,那么在前缀和序列中呢?

        我们通过观察发现 14 = 15 - 1,15所在位置刚好为4,1所在位置为1,所以我们大胆猜测l到r的和是否为前缀和序列中 sum[r] - sum[l - 1];

               sum[4] = a[1] + a[2] + a[3] + a[4], sum[2 - 1] = a[1],

                在原数组中为:a[2] + a[3] + a[4];

        由上可以看出我们猜测是对的。

前缀和做法:

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int arr[N], sum[N];
int get_sum(int l, int r, int sum[]){
    if(l <= 1) return sum[r];
    return sum[r] - sum[l - 1];
}
int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++) cin >> arr[i];
    for(int i = 1; i <= n; i++) sum[i] = sum[i - 1] + a[i];
    while(m--){
        int l, r;
        cin >> l >> r;
        int ans = get_sum(l, r, sum);    
        cout << ans << endl;
    } 
    return 0;
}

        这样一来我们代码的时间复杂度就变成了O(n)了。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值