線段樹 hate it

nearal發表於2020-09-27

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
對於每一次詢問操作,在一行裡面輸出最高成績。
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>
#include<algorithm>
using namespace std;
int tree[1001000],n,m;
void build(int L,int R,int idx){
	if(L==R){    //遞迴到葉節點 
		scanf("%d",&tree[idx]);
		return ;
	}
	int mid=(L+R)>>1;
	build(L,mid,idx<<1);  //遞迴構建左子樹 
	build(mid+1,R,idx<<1|1);  //遞迴構建右子樹 
	tree[idx]=max(tree[idx<<1],tree[idx<<1|1]);
}
void update(int x,int y,int L,int R,int idx){
	if(L==R&&L==x){  //判斷當前節點是否在更新區間內
		tree[idx]=y;
		return ;
	}
	int mid=(L+R)>>1;
	if(x<=mid)
	    update(x,y,L,mid,idx<<1);//回溯更新左子樹 
	else
	    update(x,y,mid+1,R,idx<<1|1);//回溯更新到右子樹 
	tree[idx]=max(tree[idx<<1],tree[idx<<1|1]); //更新父節點 
}
int query(int x,int y,int L,int R,int idx){
	if(x<=L&&y>=R){  //判斷當前節點是否在查詢區間內 
		return tree[idx];
	}
	int mid=(L+R)>>1;
	int s=0,ss=0;
	if(x<=mid)
	    s=query(x,y,L,mid,idx<<1);
	if(y>mid)
	    ss=query(x,y,mid+1,R,idx<<1|1);
	return max(s,ss);//查詢左右子樹,並返回值 
}
int main(){
	int a,b;
	char c;
	while(~scanf("%d%d",&n,&m)){
		build(1,n,1);		
		while(m--){
			getchar();
			scanf("%c",&c);
			scanf("%d%d",&a,&b);
			if(c=='Q')
			    printf("%d\n",query(a,b,1,n,1));
			else if(c=='U')
			    update(a,b,1,n,1);
		}
	}
	return 0;
} 

相關文章