LibreOJ 131 树状数组2:区间修改,单点查询

题目地址

Solution

前面已经知道了树状数组的单点修改和区间查询。这里利用差分的思想:具体来说,维护 \(b\) 数组:

\[b[i] = a[i]-a[i-1] \]

其中 \(a\) 为原来数组。可以发现

\[a[i] = \sum_{k=1}^ib[k] \]

因此我们只需要对 \(b\) 利用树状数组维护,get_sum[i]即可得到原来数组单点的值。而对于区间更新,我们只需要在两个端点 \(l,r\) 打上标记即可:

\[update(l,x);update(r+1,-x) \]

点击查看代码
 int n,q; const int N = 1e6+5; ll a[N], b[N]; ll c[N];  ll lowbit(int x){return x&(-x);}  void update(int i, ll k){     // add k on i-th position     while(i<=N){         c[i]+=ll(k); i+= lowbit(i);     } } ll get_sum(int i){     // get sum from 1-i     ll ans = 0;     while(i>0){         ans+=ll(c[i]);i-=lowbit(i);     }     return ans; }  int main(){     //ios::sync_with_stdio(false);     n = read();q = read();     for(int i=1;i<=n;i++){         scanf(%lld,&a[i]);         b[i] = a[i]-a[i-1];         update(i,b[i]);     }     while(q--){         int tmp = read();         if(tmp==1){             int l=read(),r = read();             ll x;scanf(%lld,&x);             update(l,x);update(r+1,-x);         }         else{             int pos = read();             cout<<get_sum(pos)<<endl;         }     }      }