这个是我的第一道线段树。
总之,看过理解了二分搜索树的,基本上,看这个线段树,没什么难度的。
把一个区间分成左右两个儿子。左儿子比父亲区间的中点小,右二子大。
比如,[1,10]为父亲节点,那么左儿子节点为[1,5],右儿子为[6,10].
这样每个节点保存相应的数据。比如,此题保存为这段区间的最大分数。
总的时间为O(logN).
#include <iostream>
using namespace std;
int n,m;
int ts,te;
int mg;
int sgTree[530000]; //线段树的数组形式.i*2为左儿子节点,i*2+1为右儿子节点
int max(int a,int b)
{
return a>b?a:b;
}
void query(int s,int e,int c,int ms,int me)
{
if(s==ms&&e==me) //如果恰好是在[s,e]这个区间,那么记录下来最大成绩。
{
mg=max(mg,sgTree[c]);
return;
}
int mid=(s+e)/2;
if(mid>=me) query(s,mid,c*2,ms,me); //如果要搜索的区间的最大值不比mid大。那么,进入左儿子搜索
else if(mid<ms) query(mid+1,e,c*2+1,ms,me); //同理
else
{
query(s,mid,c*2,ms,mid); //这里,如果在左右儿子都有区间的话,那么就把[ms,me]也分成两个区间,分别搜索。
query(mid+1,e,c*2+1,mid+1,me);
}
}
void update(int s,int e,int c)
{
if(s==e&&s==ts)
{
sgTree[c]=te;
return;
}
int mid=(s+e)/2;
if(mid>=ts) update(s,mid,c*2);
if(ts>mid) update(mid+1,e,c*2+1);
sgTree[c]=max(sgTree[c*2],sgTree[c*2+1]);
}
void createTree(int s,int e,int c)
{
if(s==e) //如果这个节点左右区间相等,那么就是一个学生,此时,输入成绩
{
scanf("%d",&sgTree[c]);
return ;
}
int mid=(s+e)/2; //mid为此区间的中点
createTree(s,mid,c*2); //创建左儿子节点
createTree(mid+1,e,c*2+1); //创建右儿子节点
sgTree[c]=max(sgTree[c*2],sgTree[c*2+1]); //这里采用后序遍历。因为只有,左右儿子都创建并且知道它们的最大成绩,这里才能比较并存储较大的成绩
}
int main()
{
char ch[2];
while(scanf("%d %d ",&n,&m)==2)
{
createTree(1,n,1);
for(int i=0;i<m;i++)
{
scanf("%s %d %d",&ch,&ts,&te); //这里采用字符串输入,开始老是%c输入不正确,干脆就用字符串了。
mg=0;
if(ch[0]=='Q')
{
query(1,n,1,ts,te);
printf("%d\n",mg);
}
if(ch[0]=='U')
update(1,n,1);
}
}
return 0;
}