概念:
线段树(Segment Tree)由n个包含了1段区间(线段)的完全二叉树组成的树形数据结构。
思想:
利用它的树形结构,将其包含的数据进行快速的修改和查询。(可用于维护dp)
基本操作:
存储方式:
struct Tree{
int left,right,value; //left表示左端点,right表示右端点,value表示存储的值
}tree[MAXN*4+1]; //*2是树形结构需要比原序列大一倍,再*2是为了省去特判
/*
int left[MAXN*4+1],right[MAXN*4+1],value[MAXN*4+1];//同上
*/
建树:
利用完全二叉树的性质,用递归的思想,由根节点开始逐层递归至叶子节点。
void make_tree(int l,int r,int p) { //l表示当前线段的左区间,r表示当前线段的右区间,p表示该线段的编号
tree[p].left=l,tree[p].right=r;
if(l==r) {
tree[p].value=a[l]; // or tree[p].value=a[r];
return ;
}
int mid=(l+r)>>2; //枚举中点
make_tree(l,mid,p<<1); //向左递归
make_tree(mid+1,r,p<<1|1); //向右递归
tree[p].value=tree[p<<2].value+tree[p<<2+1].value; //父节点等于左节点加右节点
//如果是区间查找最值,则将它改成max即可,下同
}
单点修改:
void add(int l,int r,int x,int y,int p) { //l表示线段的左端点,r表示线段的右端点,x表示要修改的数,y表示修改后的数,p表示节点的编号
if(x<l || x>r) return ; //如果要修改的数不在当前区间则返回
if(l==r) { //如果左端点等于右端点,即l=r=x,则对x的值进行修改
tree[p].value=y;
return ;
}
int mid=(l+r)>>1; //枚举中点
add(l,mid,x,y,p<<1); //更新左子树
add(mid+1,r,x,y,p<<1|1); //更新右子树
tree[p].value=tree[p<<1].value+tree[p<<1|1].value;
}
区间查询累加和:
int query(int l,int r,int p) { // l表示待查询区间的左端点,r表示带查询区间的右端点
if(tree[p].left==l && tree[p].right==r) return tree[p].value; //如果当前区间包含了待查询的区间则返回当前区间的值
int mid=(l+r)>>1;
if(r<=mid) return query(l,mid,p<<1); //如果待查询的区间位于中点的左侧,则向左递归
else if(l>mid) return query(mid+1,r,p<<1|1); //如果待查询的区间位于中点的右侧,则向右递归
else return query(l,mid,p<<1)+query(mid+1,r,p<<1|1); //如果待查询的区间部分位于中点左侧,部分位于中点右侧,则两边都需要递归
//如果是区间查询最值,将query(l,mid,p<<1)+query(mid+1,r,p<<1|1)改成max(query(l,mid,p<<1),query(mid+1,r,p<<1|1))即可
}