一个简单的整数问题

题目描述

给定长度为N的数列A，然后输入M行操作指令。
第一类指令形如“C l r d”，表示把数列中第l~r个数都加d。
第二类指令形如“Q X”，表示询问数列中第x个数的值。
对于每个询问，输出一个整数表示答案。

输入格式

第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。
每个答案占一行。

数据范围

1≤N,M≤105,
|d|≤10000,
|A[i]|≤1000000000

输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2
输出样例：
4
1
2
5

题目分析

将[l,r]区间中的数全部加上d，这个操作我们可以想到用差分来完成。
但是除了这个操作之外还有第二个操作：获得第x个数的值。
因此不能简单的用只差分，我们可以用树状数组来维护一个差分数组，这样就可以快速的完成上面的两个操作了。

代码如下

#include <iostream>
#include <cstring>
#define LL long long
using namespace std;
const int N=1e5+5;
int n,m;
int a[N];
LL tri[N];
int lowbit(int x)
{
    return x & -x;
}
void add(int x,int c)
{
    for(int i=x;i<=n;i+=lowbit(i))
        tri[i]+=c;
}
LL sum(int x)
{
    LL res=0;
    for(int i=x;i;i-=lowbit(i))
        res+=tri[i];
    return res;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        add(i,a[i]-a[i-1]);		//因为是差分数组，所以每次插入应提前处理成差分的数
    }
    while(m--)
    {
        char op[2];
        int l,r,d;
        scanf("%s %d",op,&l);
        if(op[0]=='C')
        {
            scanf("%d %d",&r,&d);
            add(l,d); add(r+1,-d);		//用差分操作来加数即可
        }
        else printf("%lld\n",sum(l));	//x位置上的数即为tri[]中x的前缀和
    }
    return 0;
}