欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-11 17:05:32
...

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次查询的计算结果。

Input示例

5
1
3
7
9
-1
4
1 2
2 2
3 2
1 5

Output示例

4
10
16
19

树状数组详解

代码:

#include <iostream>
#define ll long long
using namespace std;
ll c[50005];
int n;
int lowbit(int x) {
	return (-x)&x;
}
ll sum(int k) {
	ll s = 0;
	while(k > 0) {
		s += c[k];
		k -= lowbit(k);
	}
	return s;
}
void update(int i, int x) {
	while(i <= n) {
		c[i] += x;
		i += lowbit(i);
	}
}
int main() {
	int x;
	cin>>n;
	for(int i = 1; i <= n; i++) {
		cin>>x;
		update(i, x);
	}
	int m, a, b;
	cin>>m;
	while(m--) {
		cin>>a>>b;
		cout<<sum(a+b-1)-sum(a-1)<<endl;
	}
	return 0;
}

 

相关标签: 树状数组