HDOJ H1754 I Hate It（结构体和数组模拟线段树）



http://acm.hdu.edu.cn/showproblem.php?pid=1754

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 44810    Accepted Submission(s): 17587

Problem Description

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

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

 

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

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

 

Sample Input

  
  

   
   5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5
  
  

 

Sample Output

  
  

   
   5
6
5
9

  
  

#include<stdio.h>
#include<string.h>
#define MAX(a,b) (a>b?a:b)
#define MIN(a,b) (a<b?a:b)
#define node 200000+10
//---------------------------------------------------------------------------------------//
#define MAX_node 600000+10 //叶子节点为n的线段树的节点数为(2-1/2^(n-1))*n，这里0<n<=20000 
//叶子节点为n，节点最多（满二叉树）为Sn=n+n/2+2/4+...+n/n=(1+1/2+1/3+...+1/n)*n=(2-1/2^(n-1))*n 
//由底数为1/2的指数函数的图像可知 0<Sn<2*n。应该是400000就可以了，不知为什么600000才AC 
//---------------------------------------------------------------------------------------//
struct Node
{//各个节点 
	int max;
	int lchild,rchild;
}num[MAX_node];
int father[node],max;// 每个叶子节点对应的结构体下标
void CreatTree(int head,int a,int b) 
{
	num[head].lchild=a;
	num[head].rchild=b;
	num[head].max=-1;//每个节点初始化都为-1
	
	if(a==b)//叶子节点
	{
		father[a]=head;
		return; 
	}
	CreatTree(head<<1,a,(a+b)/2);
	CreatTree((head<<1)+1,(a+b)/2+1,b);
}
void UpdateTree(int t)
{
	if(t==1) return;
	int f=t>>1;//f是t的父节点的标号 
	num[f].max=MAX(num[f<<1].max,num[(f<<1)+1].max);//f节点的值为左右孩子的最大值
	UpdateTree(f); 
}
void Find(int h,int A,int B)
{
	if(num[h].lchild==A&&num[h].rchild==B)
	{
		max=MAX(max,num[h].max);
		return;
	}
	h=h<<1;
	if(A<=num[h].rchild)
	{
		if(B<=num[h].rchild)
		Find(h,A,B);
		else
		Find(h,A,num[h].rchild);
	}
	h+=1;
	if(B>=num[h].lchild)
	{
		if(A>=num[h].lchild)
		Find(h,A,B);
		else
		Find(h,num[h].lchild,B);
	}
}
int main()
{
	int N,M,n,i,A,B;
	char c;
	while(~scanf("%d %d",&N,&M))
	{
		memset(father,0,sizeof(father));
		CreatTree(1,1,N);//为区间[1,N]建立一个以1为祖先的线段树.
		for(i=1;i<=N;i++)
		{//对每个节点赋值 
			scanf("%d",&n);//初始化i叶子节点的值为n
			num[father[i]].max=n; 
			UpdateTree(father[i]);
		}
		for(i=0;i<M;i++)
		{
			getchar();
			scanf("%c %d %d",&c,&A,&B);
			if(c=='Q')//输出A~B之间的最大值
			{
			    max=-1;
			    Find(1,A,B);//从根开始往下找A~B之间最大的数
		        printf("%d\n",max);
			}
			if(c=='U')//将A节点的值改为B
			{
			   num[father[A]].max=B;
			   UpdateTree(father[A]);
		    }
		}
    }
	return 0; 
}