51NOD 1081 子段求和

1081 子段求和

 

给出一个长度为N的数组，进行Q次查询，查询从第i个元素开始长度为l的子段所有元素之和。

 

例如，1 3 7 9 -1，查询第2个元素开始长度为3的子段和，1 {3 7 9} -1。3 + 7 + 9 = 19，输出19。

Input

第1行：一个数N，N为数组的长度(2 <= N <= 50000)。
第2 至 N + 1行：数组的N个元素。(-10^9 <= N[i] <= 10^9)
第N + 2行：1个数Q，Q为查询的数量。
第N + 3 至 N + Q + 2行：每行2个数，i，l（1 <= i <= N，i + l <= N)

Output

共Q行，对应Q次查询的计算结果。

前缀和

#include <bits/stdc++.h>
using namespace std;
const int maxn = ;
typedef long long ll;
int n,k;
int s[maxn];

ll sum [maxn];

int main (){

    scanf("%d ", &n);
    for(int i=; i <= n; i++)
        scanf("%d",&s[i]);
    sum[] = ;
    for(int i=;i <= n;i++){
        sum [i] = sum[i-] + s[i];
    }

    scanf("%d", &k);

    for(int i=;i <= k;i++){
        int a,b;
        scanf("%d %d",&a,&b);
        printf("%lld\n",sum[a+b-] - sum[a-]);
    }
    return ;
}

树状数组

#include <bits/stdc++.h>
using namespace std;
const int maxn = ;
typedef long long ll;
int n,k;

ll bit[maxn];

void update(int i,int d){
    while (i <= n){
        bit[i] += d;
        i += i & (-i);
    }
}

ll sum (int i){
    ll s = ;
    while (i > ){
        s += bit[i];
        i -= i & (-i);
    }
    return s;
}

int main (){
    memset(bit,,sizeof(bit));
    scanf("%d", &n);    int x;
    for(int i=; i <= n; i++){
        scanf("%d",&x);
        update(i,x);
    }

    scanf("%d", &k);
    for(int i=;i <= k;i++){
        int a,b;
        scanf("%d %d",&a,&b);
        printf("%lld\n",sum(a+b-) - sum(a-));
    }
    return ;
}