HDU_1754 I Hate It [线段树模板]

题目描述

很多学校流行一种比较的习惯。老师们很喜欢询问，从某某到某某当中，分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢，现在需要你做的是，就是按照老师的要求，写一个程序，模拟老师的询问。当然，老师有时候需要更新某位同学的成绩。

Input

本题目包含多组测试，请处理到文件结束。
在每个测试的第一行，有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 )，分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数，代表这N个学生的初始成绩，其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取’Q’或’U’) ，和两个正整数A，B。
当C为’Q’的时候，表示这是一条询问操作，它询问ID从A到B(包括A,B)的学生当中，成绩最高的是多少。
当C为’U’的时候，表示这是一条更新操作，要求把ID为A的学生的成绩更改为B。

Output

对于每一次询问操作，在一行里面输出最高成绩。

线段树，用树的方式维护区间的特性，在面对多次修改，和查询时可以达到很高的速度。（详见代码）

#include<cstdio>//线段树的模板，线段树是一个二叉搜索树，其实就是二叉树 
#include<iostream>//只不过每个节点保存的是其对应区间的值，这里是最大值,父节点区间是子节点区间的并集 
#include<algorithm>
#define pointnumber 200005//定义常量表示总区间的长度 
using namespace std;

struct node//用结构体表示线段树的节点 
{
	int va,le,ri;//va表示值,le,ri表示左右端点,闭区间 
}tree[pointnumber*4+1];//我们的线段树，大小应该是总长的四倍 
int origin[pointnumber];//原始数据保存在这里 
int built(int lef,int rig,int num)//用来建立线段树的函数 
{//lef,rig是左右端点的值，num代表当前节点的下标 
	tree[num].le=lef;
	tree[num].ri=rig;//该节点的左右端点进行赋值 
	if(lef==rig)//左端点等于右端点，说明这是一个不可再分只包含一个数的区间 
	{//该区间的最大值就是这个唯一的数 
		tree[num].va=origin[lef];
		return tree[num].va;//返回最大值，供父节点判断父节点自身的最大值 
	}
	int min=(lef+rig)>>1;//将区间尽可能的对半分 
	
	tree[num].va=max(built(lef,min,num<<1),built(min+1,rig,(num<<1)+1));
	//大区间的最大值是左右区间最大值较大的一个
	//也可以对这里进行修改变为最小、求和...等等 
	return tree[num].va; 
}
int pointask(int k,int num)//访问某个点的值 
{
	if(tree[num].le==tree[num].ri)//点的位置都在叶节点 说明找到了要访问的点 
	{
		return tree[num].va;//返回该点的值 
	}
	int min=(tree[num].le+tree[num].ri)>>1;//计算中间点位置，也是左结点的右端点 
	if(k<=min)
	{
		return pointask(k,num<<1);//如果在左边 对左边进行递归 
	}
	return pointask(k,(num<<1)+1);//对右边进行递归 
}
int repoint(int k,int num,int v)//修改某个点的值，并维护树状数组 
{
	if(tree[num].le==tree[num].ri)//找到要求的点 
	{
		tree[num].va=v;//修改它的值 
		return tree[num].va;
	}
	int min=(tree[num].le+tree[num].ri)>>1;//中点 
	int val1,val2;//用于保存左右子节点的va值 
	if(k<=min)//在左侧 
	{//右边的值不用变 对左子树维护 
		val1=repoint(k,num<<1,v);
		val2=tree[(num<<1)+1].va;
	}
	else //在右侧 
	{//对右边递归 
		val1=tree[num<<1].va;
		val2=repoint(k,(num<<1)+1,v);
	}
	tree[num].va=max(val1,val2);
	return tree[num].va;
}
int partask(int l,int r,int num)//访问区间的值 
{
	int min=(tree[num].le+tree[num].ri)>>1;//中点
	
	if(tree[num].le==l&&tree[num].ri==r)
	{//如果查询的区间刚好等于当前节点的区间 
		return tree[num].va;
	}//直接返回va 
	
	if(r<=min) //如果查询的区间在左侧 
	{
		return partask(l,r,num<<1);
	}
	if(l>=min+1)//在右侧 
	{
		return partask(l,r,(num<<1)+1);
	}
	//一半在左一半在右 分成两半 
	return max(partask(l,min,num<<1),partask(min+1,r,(num<<1)+1));
} 
int main()
{
	int n,m,i,j,x,y,ans;
	char ch; 
	while(scanf("%d%d",&n,&m)!=EOF)
	{//printf("\n*******\n");
		
		
		for(i=1;i<=n;i++)
		{
			scanf("%d",&origin[i]);
		}
		
		built(1,n,1);
		for(i=1;i<=m;i++)
		{
			scanf(" %c",&ch);
			scanf("%d%d",&x,&y);
			if(ch=='Q')
			{
				ans=partask(x,y,1);
				printf("%d\n",ans);
			}
			else if(ch=='U')
			{
				repoint(x,1,y);
			}
		}
	} 
	return 0;
}