[51Nod](1081)子段求和 ---- 树状数组
程序员文章站
2022-05-11 17:04:56
...
给出一个长度为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
思路: 一开始做着题,还是半年前辣么“暴力”的我,没想到o(n^2) 984ms 就AC了,毕竟数据也小。
当然懂了树状数组,就去解决这种直白的区间求和问题~
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 50005;
typedef long long ll;
int n,t;
ll a;
ll c[maxn];
int lowbit(int x)
{
return x&(-x);
}
ll getsum(int e)
{
ll ans = 0;
for(int i=e;i>0;i-=lowbit(i))
ans+=c[i];
return ans;
}
void update(int st,int x)
{
for(int i=st;i<=n;i+=lowbit(i))
{
c[i]+=x;
}
}
int main()
{
#ifdef LOCAL
freopen("in.txt","r",stdin);
#endif // LOCAL
ios_base::sync_with_stdio(false);
cin.tie(NULL),cout.tie(NULL);
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a;
update(i,a);
}
cin>>t;
while(t--)
{
int s,l;
cin>>s>>l;
cout<<getsum(s+l-1)-getsum(s-1)<<endl;
}
return 0;
}
上一篇: php socket通信简单实现
下一篇: 自定义消息通知、手机震动、提示音