RMQ问题——线段树+懒标记
线段树，基于分治思想，用来维护区间信息的二叉树结构
例如RMQ，区间和，区间GCD问题，在平均 \(O(\log n)\) 的时间复杂度内执行区间修改和查询操作
朴素的线段树的每个节点包含三个元素：左区间，右区间，区间元素统计值（以区间和为例）
struct node{int l,r,sum;
};
对于父节点的编号 \(p\) ，左右孩子的编号即为 \(2*p,2*p+1\)
#define lc p<<1;//定义左儿子
#define rc p<<1|1//定义右儿子
建树时，采用递归的方式，自下而上赋值
node tr[4*N];
void build(int p,int l,int r){tr[p]={l,r,w[l]};//递归到底层，给叶子节点赋值，非叶子节点无效if (l==r) return;//递归出口int m=l+r>>1;//二分树build(lc,l,m);//递归建树左儿子build(rc,m+1,r);//递归建树右儿子tr[p].sum=tr[lc].sum+tr[rc].sum;//递归回溯，赋值非叶子节点的区间统计值
}
有关线段树为什么需要开 \(4*N\) 空间的问题：
因为线段树不一定为满二叉树，所以对于区间 \([1,n]\) ，底层的叶子节点数最差情况为 \(2^{\lfloor\log _2n\rfloor+1}\)，所以总节点数最差为
\[2^{\lfloor\log _2n\rfloor+1}*2-1=4*n-1\approx4*n
\]
针对区间某一点进行修改，采用递归的方式从树顶进入，到达目标叶子节点然后回溯更新非叶子节点
void update(int p,int x,int k){//p一开始为1，即从根节点进入 x为目标点，k为需要修改为的值if (tr[p].l==x&&tr[p].r==x){//到达需要修改的点tr[p].sum=k;//更新值return;}int m=tr[p].l+tr[p].r>>1;//分治if (x<=m) update(lc,x,k);//x在左半边else update(rc,x,k);//x在右半边tr[p].sum=tr[lc].sum+tr[rc].sum;//回溯更新非叶子节点的区间统计值
}
对某一区间进行统计值的查询，采取拆分和拼凑的办法
- 如果二分下来的区间能被需要查询的区间完全覆盖，那么直接加上这个区间的统计值即可（拼凑答案的一部分）
- 左儿子与需要查询的区间有重合，那就递归左子树
- 右儿子与需要查询的区间有重合，那就递归右子树
int query(int p,int x,int y){if (x<=tr[p].l&&y>=tr[p].r)//能完全覆盖的区间一定是答案的一部分return tr[p].sum;int m=tr[p].l+tr[p].r>>1;int sum=0;if (x<=m) sum+=query(lc,x,y);//访问左子树if (y>m) sum+=query(rc,x,y);//访问右子树return sum;
}
懒惰修改：
修改区间时给非叶子节点添加懒标记，只在必要的条件下下传懒标记，否则一直维持在非叶子节点上
这个操作可以将朴素的区间修改操作的 \(O(n)\) 优化到 \(O(\log n)\)
struct node{int l,r,sum,add;//额外增添懒标记
};
void build(int p,int l,int r){tr[p]={l,r,w[l],0};//建树时，初始化懒标记为0...
}
下传懒标记，区间修改操作：
void pushdown(int p){//下传懒标记if (tr[p].add){tr[lc].sum+=tr[p].add*(tr[lc].r-tr[lc].l+1);//更新左儿子的统计值tr[rc].sum+=tr[p].sum*(tr[rc].r-tr[rc].l+1);//更新右儿子的统计值tr[lc].add+=tr[p].add;//下传给左儿子tr[rc].add+=tr[p].add;//下传给右儿子tr[p].add=0;}
}void update(int p,int x,int y,int k){//更新xy区间的值都增加kif (x<=tr[p].l&&y>=tr[p].r){//如果完全覆盖，直接修改非叶子节点统计值，添加懒标记tr[p].sum+=(tr[p].r-tr[p].l+1)*k;tr[p].add+=k;return;}int m=tr[p].l+tr[p].r>>1;pushdown(p);//下传懒标记if (x<=m) update(lc,x,y,k);//维护左子树if (y>m) update(rc,x,y,k);//维护右子树tr[p].sum=tr[lc].sum+tr[rc].sum;//更新非叶子节点区间统计值
}
同样的，添加了懒标记后，区间查询操作仍然需要下传
int query(int p,int x,int y){if (x<=tr[p].l&&y>=tr[p].r)//能完全覆盖的区间一定是答案的一部分return tr[p].sum;int m=tr[p].l+tr[p].r>>1;pushdown(p);//下传懒标记int sum=0;if (x<=m) sum+=query(lc,x,y);//访问左子树if (y>m) sum+=query(rc,x,y);//访问右子树return sum;
}
