【洛谷 p3368】模板-树状数组 2（数据结构）

题目：已知一个数列，你需要进行下面两种操作：1.将某区间每一个数数加上x；2.求出某一个数的和。

解法：树状数组+前缀和优化。数组中每位存和前一位的数的差，这样区间修改只用改两位，单点询问就是求前缀和。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;

const int N=500010;
int n;
int c[N];

int lowbit(int x) {return x&-x;}
void ins(int x,int d)
{
    for (int i=x;i<=n;i+=lowbit(i))
      c[i]+=d;
}
int query(int x)
{
    int h=0;
    for (int i=x;i>=1;i-=lowbit(i))
      h+=c[i];
    return h;
}
int main()
{
    int m,x,y,k;
    scanf("%d%d",&n,&m);
    y=0;
    for (int i=1;i<=n;i++)
    {
      scanf("%d",&x);
      ins(i,x-y);
      y=x;
    }
    while (m--)
    {
      scanf("%d",&k);
      if (k==1)
      {
        scanf("%d%d%d",&x,&y,&k);
        if (x>y) {int t;t=x,x=y,y=t;}
        ins(x,k),ins(y+1,-k);
      }
      else
      {
        scanf("%d",&x);
        printf("%d
",query(x));
      }
    }
    return 0;
}