~这是哪一题我不记得了~
zkw(重口味)线段树,是一个神奇的东西。
它可以支持区间快速查询和修改的操作。
现在我来介绍一下这个神奇的线段树吧!
(这个神奇的线段树是从叶子节点开始的哦~)
题意:
有n个数,m个操作,
C是查询x到y区间的和,
U是修改区间的第u个点成v;
1.create_tree(建树):
这是对于满二叉树来说的。
所以叶子节点的起点的编号为2^(n-1)+1,//n为二叉树的层数
利用二叉树的性质,可以快速的找到。
再依次将每个叶子节点输入,
维护二叉树。
~这和普通的二叉树不同~
void create_tree(int n){
for(M=1;M<=n+1;M<<=1); //找到第一个叶子节点
for(int i=M+1;i<=M+n;i++)scanf("%d",&tree[i]);//输入叶子节点
for(int i=M-1;i>0;i--)push_up(i);//维护叶子节点的父亲节点
}
2.查询:
(这里的位运算可能会有点恶心,但也不过多的讲述,可以百度~)
找要查询区间的左边的那个点,记录它左边的的左边的那一个点。
同时也找到右边的点,记录它右边的右边的那个点。
~这也利用了二叉树左子树和右子树的性质~
int query(int s,int t){
int ans=-1;
for(s=s+M-1,t=t+M+1;s^t^1;s>>=1,t>>=1){
if(~s&1)ans=max(ans,tree[s^1]);
if(t&1)ans=max(ans,tree[t^1]);
}
}
3.修改:
直接修改某个叶子结点,
维护区间。
void update(int n,int v){
for(tree[n+M]=v,n>>=1;n;n>>=1)push_up(n);//直接修改叶子结点
}
总程序:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=200010;
int tree[maxn*4],M;
void push_up(int u){tree[u]=max(tree[u<<1|1],tree[u<<1]);}
void create_tree(int n){
for(M=1;M<=n+1;M<<=1); //找到左子节点
for(int i=M+1;i<=M+n;i++)scanf("%d",&tree[i]);//输入叶子节点
for(int i=M-1;i>0;i--)push_up(i);//维护叶子节点的父亲节点
}
int query(int s,int t){
int ans=-1;
for(s=s+M-1,t=t+M+1;s^t^1;s>>=1,t>>=1){
if(~s&1)ans=max(ans,tree[s^1]);
if(t&1)ans=max(ans,tree[t^1]);
}
}
void update(int n,int v){for(tree[n+M]=v,n>>=1;n;n>>=1)push_up(n);}
int main(){
int m,n;
while(scanf("%d%d",&n,&m)){//输入
create_tree(n);//建树
for(int i=1;i<=m;i++){
char s[15];int u,v;//省略了getchar
scanf("%s%d%d",s[i],&u,&v);
if(s[0]=='U')update(u,v);
else printf("%d\n",query(u,v));
}
}
return 0;
}