给出一个长度为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 1619
解题思路:裸的树状数组。
代码如下:
#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; }