字段求和

基准时间限制:1 秒 空间限制:131072 KB 分值: 0  难度:基础题
 收藏
 关注
给出一个长度为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次查询的计算结果。
Input示例
5
1
3
7
9
-1
4
1 2
2 2
3 2
1 5
Output示例
4
10
16
19

可以用线段树也可以用树状数组求。

#include<map>
#include<stack>
#include<queue>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#define maxn 50005
#define inf 0x3f3f3f
#define ll  long long
#define mod 1000000000
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
ll c[maxn];
int n;
int lowbit(int x){
    return x&(-x);
}
void add(int x,int y){
    for(int i=x;i<=n;i+=lowbit(i)){
        c[i]+=y;
    }
}
ll getsum(int x){
    ll ans=0;
    while(x>0){
        ans+=c[x];
        x-=lowbit(x);
    }
    return ans;
}
int main(){
    while(cin>>n){
        mem(c,0);
        for(int i=1;i<=n;i++){
            ll x;cin>>x;
            add(i,x);//add(i+1,-x);
        }
        int q;cin>>q;
        for(int i=0;i<q;i++){
            int a,b;cin>>a>>b;
            cout<<getsum(a+b-1)-getsum(a-1)<<endl;
        }
    }
}

猜你喜欢

转载自blog.csdn.net/lpeaceminusone/article/details/81046140
今日推荐