【模板】树状数组的区间加值和区间查询
2017-12-11 09:40
387 查看
树状数组可以实现单点修改区间查询,或是差分的区间修改单点查询。【点这里】
那么树状数组能不能实现区间加值区间查询呢?
原始序列为a[]。我们先考虑区间加值,用差分的方式维护一个d[]数组,d[x]表示的是[x,maxn]每个数的增量。
那么如果需要查询1~x的前缀和,需要查的是
sumx=sigma(ai)+d1∗x+d2∗(x−1)+d3∗(x−2)+...+dx∗1
=sigma(ai)+sigma(di∗(x+1−i))
=sigma(ai)+(x+1)∗sigma(di)−sigma(di∗i)
所以把di处理一个树状数组tree[],di∗i处理成一个树状数组treex[]。查询sum[x]时只需要求suma[x]+(x+1)*tree[x]-treex[x]就可以了。
模板题是【模板】线段树1 洛谷P3372
那么树状数组能不能实现区间加值区间查询呢?
原始序列为a[]。我们先考虑区间加值,用差分的方式维护一个d[]数组,d[x]表示的是[x,maxn]每个数的增量。
那么如果需要查询1~x的前缀和,需要查的是
sumx=sigma(ai)+d1∗x+d2∗(x−1)+d3∗(x−2)+...+dx∗1
=sigma(ai)+sigma(di∗(x+1−i))
=sigma(ai)+(x+1)∗sigma(di)−sigma(di∗i)
所以把di处理一个树状数组tree[],di∗i处理成一个树状数组treex[]。查询sum[x]时只需要求suma[x]+(x+1)*tree[x]-treex[x]就可以了。
模板题是【模板】线段树1 洛谷P3372
#include<bits/stdc++.h> #define lowbit(x) (x&(-x)) using namespace std; int n,m; long long a[100005]; long long tree[100005],treex[100005]; inline void add(long long *ths,int x,long long num){ while(x<=n){ ths[x]+=num; x+=lowbit(x); } } inline long long search(long long *ths,int x){ long long re=0; while(x){ re+=ths[x]; x-=lowbit(x); } return re; } inline long long search(int x){ return a[x]+(x+1)*search(tree,x)-search(treex,x); } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;++i) scanf("%lld",&a[i]),a[i]+=a[i-1]; while(m--){ int opt; scanf("%d",&opt); if(opt==1){ int l,r; long long k; scanf("%d%d%lld",&l,&r,&k); add(tree,l,k); add(tree,r+1,-k); add(treex,l,l*k);add(treex,r+1,-(r+1)*k); } else{ int l,r; scanf("%d%d",&l,&r); printf("%lld\n",search(r)-search(l-1)); } } return 0; }
相关文章推荐
- P3368 【模板】树状数组 2(区间增减,单点查询)
- [模板]-树状数组③-区间修改+区间查询
- [模板]树状数组(区间修改单点查询)
- CodeVS 1082 (区间修改+区间查询树状数组模板)
- P3368 【模板】树状数组 2 单点查询与区间修改
- hdu1754 I hate it线段树模板 区间最值查询
- hdu 4970 Killing Monsters 【树状数组(区间修改,点查询)】
- HDU - 4777 Rabbit Kingdom 预处理 + 树状数组(区间更新点查询)
- 树状数组 --区间查询+区间修改
- HDU 1556 Color the ball(线段树区间更新区间查询模板)
- (模板)树状数组 (区间修改,单点查询)
- hdu p1556 树状数组的区间修改,单点查询。
- ACM常用模板——数据结构——区间最值查询RMQ
- SPOJ3267 D-Query 树状数组离线操作 或 主席树 查询某一区间内有多少不同的数
- HDU1823 Luck and Love(二维线段树单点更新+区间查询+模板)
- 线段树模板(点修改 ,区间查询)
- 树状数组模板1——单点修改区间查询
- 4000 <模板>区间更新,单点查询
- A Simple Problem with Integers POJ - 3468 (线段树区间增减,区间查询模板)
- Splay 区间信息的维护与查询 静态模板