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\) 打上标记即可:
点击查看代码
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; } } }