基准时间限制: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<iostream>
#include<stdio.h>
using namespace std;
long long a[500005];
int n;
int lowbit(int i)
{
return i&(-i);
}
long long sum(int l)
{
long long sum=0;
while(l>0)
{
sum+=a[l];
l-=lowbit(l);
}
return sum;
}
void built(int i,int x)
{
while(i<=n)
{
a[i]+=x;
i+=lowbit(i);
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
int m;
cin>>m;
built(i,m);
}
int l;
cin>>l;
while(l--)
{
int a,b;
cin>>a>>b;
cout<<sum(a+b-1)-sum(a-1)<<endl;
}
return 0;
}
树状数组