您的位置:首页 > 其它

51nod:1081 子段求和(树状数组)

2017-04-18 18:40 197 查看
1081 子段求和


基准时间限制: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 <cstdio>
long long c[50010];
long long n,d;
void update(long long i,long long d)
{
while(i<=n)
{
c[i]=c[i]+d;
i=i+(i&(-i));
}
}
long long sum(long long i)
{
long long tmp=0;
while(i>0)
{
tmp=tmp+c[i];
i=i-(i&(-i));
}
return tmp;
}
int main()
{
scanf("%lld",&n);
for(long long i=1;i<=n;i++)
{
scanf("%lld",&d);
update(i,d);
}
long long q;
scanf("%lld",&q);
long long s,l;
while(q--)
{
scanf("%lld%lld",&s,&l);
printf("%lld\n",sum(s+l-1)-sum(s-1));
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  树状数组