AcWing 242. 一个简单的整数问题（树状数组+差分 区间增加 单点查询）

本题要求用树状数组实现 区间增加 和 单点查询。

前置知识：

树状数组、前缀和与差分。

题意：

思路：

本题的指令有“区间增加”，“单点查询”。

这两个指令是 裸的差分 所能够完成的，前者O(1)，后者O(n)，不过根据题目，元素个数n和询问次数m的数据范围均为1e5，如果用裸的差分，完成m次单点查询需要O(n^2)，显然不合法，应当舍弃。

正确做法：树状数组+差分，或者说用差分数组来建立树状数组。

而树状数组基本指令为：“单点增加”，“区间查询”，当结合了差分数组就能够实现“区间增加”，“单点查询”了！

（差分数组的单点增加，对应原数组区间增加，差分数组查询前i项前缀和对应原数组查询a[i]）

时间复杂度：

O(mlogn)

代码：

/*将差分数组存入树状数组，原数组即为树状数组的前缀和，将问题转化为了树状数组，
原数组第x个数即为树状数组1~x的前缀和*/
#include<bits/stdc++.h>

using namespace std;
#define int long long
const int N = 1e5+10;
int n, m;
int a[N], c[N];

int lowbit(int x) {return x & (-x);}

int ask(int x){
        int ans = 0;
        while(x) {ans+=c[x], x-=lowbit(x);}
        return ans;
}

void add(int x, int y)
{
        for(; x<=n; x+=lowbit(x)) c[x]+=y;
}

signed main()
{
        cin>>n>>m;
        for(int i=1;i<=n;++i)
        {
                scanf("%lld",&a[i]);
                add(i, a[i]-a[i-1]);//用add构造差分树状数组
        }

        while(m--)
        {
                char op[2];
                int l, r, d, x;
                scanf("%s",op);
                if(*op=='C') {
                        scanf("%lld%lld%lld", &l, &r, &d);
                        add(l, d), add(r+1, -d);//用O(1)的复杂度进行区间修改
                }
                else{
                        scanf("%lld", &x);
                        printf("%lld\n", ask(x));
                }
        }

        return 0;
}