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

[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;
}