对于二叉树来说,2n是左子节点,2n+1是右子节点。
建树复杂度为O(n),更新和查询的复杂度都是O(lgn)。
要注意对一个区间进行更新的时候不需要一直更新到底,可以提高效率,需要的时候往下带就行。
#define maxn 200000
struct node{
int l, r, sum;
int mid(){
return (l+r)>>1;
}
};
node Tree[maxn<<2];
int value[maxn+10]; //从 1 开始赋值
//初始化树,根节点是1
void init_tree(int root, int l, int r){
Tree[root].l = l;
Tree[root].r = r;
if(l == r)
Tree[root].sum = value[l];
else{
init_tree(root<<1, l, (l+r)>>1);
init_tree((root<<1)+1, ((l+r)>>1) + 1, r);
Tree[root].sum = Tree[root<<1].sum + Tree[(root<<1)+1].sum;
}
}
//查找和
int query_tree(int root, int l, int r){
int m = Tree[root].mid();
if(l == Tree[root].l && r == Tree[root].r)
return Tree[root].sum;
else{
if(l > m)
return query_tree((root<<1)+1, l, r);
else if(r <= m)
return query_tree(root<<1, l, r);
else
return query_tree(root<<1, l, m) + query_tree((root<<1)+1, m+1, r);
}
}
void update_tree(int root, int idx, int v){
if(Tree[root].l == Tree[root].r)
Tree[root].sum += v;
else{
if(idx <= Tree[root].mid())
update_tree(root<<1, idx, v);
else
update_tree((root<<1)+1, idx, v);
Tree[root].sum = Tree[root<<1].sum + Tree[(root<<1)+1].sum;
}
}
出处:http://blog.csdn.net/cfarmerreally/article/details/53574139
typedef long long ll;
const int maxn=50000+10; //数组大小
int lazy[maxn<<2]; //懒惰标记
int sum[maxn<<2];
int value[maxn]; //初始值
void PushUp(int rt) //向上更新父节点
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void PushDown(int rt,int m) //向下更新
{
if(lazy[rt]) //懒惰标记,这里的语句根据更新函数而变
{
lazy[rt<<1]+=lazy[rt];
lazy[rt<<1|1]+=lazy[rt];
sum[rt<<1]+=(m-(m>>1))*lazy[rt];
sum[rt<<1|1]+=(m>>1)*lazy[rt];
lazy[rt]=0;
}
}
void build(int l,int r,int rt) //建树
{
lazy[rt]=0;
if(l==r)
{
sum[rt]=value[l];
return;
}
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
PushUp(rt);
}
void update(int L,int R,int c,int l,int r,int rt) //将L~R区间加上c,可以根据需要修改更新函数
{
if(L<=l&&R>=r)
{
lazy[rt]+=c;
sum[rt]+=c*(r-l+1);
return;
}
PushDown(rt,r-l+1);
int m=(l+r)>>1;
if(L<=m) update(L,R,c,l,m,rt<<1);
if(m<R) update(L,R,c,m+1,r,rt<<1|1);
PushUp(rt);
}
ll query(int L,int R,int l,int r,int rt) //查询L~R
{
if(L<=l&&R>=r)
return sum[rt];
PushDown(rt,r-l+1);
int m=(l+r)>>1;
ll ret=0;
if(L<=m) ret+=query(L,R,l,m,rt<<1);
if(m<R) ret+=query(L,R,m+1,r,rt<<1|1);
return ret;
}