1. 程式人生 > >線段樹——hdu1754I Hate It

線段樹——hdu1754I Hate It

cpp pri bsp clas rip 遞歸 有關 add 數組

一、題目回顧

題目鏈接:I Hate It

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<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 800005;
int arr[maxn>>1];
struct SegTreeNode{
    int val;
};
SegTreeNode segTree[maxn];

int Max(int x,int y){return x>=y ? x:y;}
 
void build(int root,int arr[],int istart,int iend)
{
    if(istart == iend)                      //葉子結點
        segTree[root].val = arr[istart];    /* 只有一個元素,節點記錄該單元素 */
    else{
        int mid = (istart + iend)/2;
        build(2*root,arr,istart,mid);       //遞歸構造左子樹
        build(2*root+1,arr,mid+1,iend);     //遞歸構造右子樹
        //根據左右子樹根節點的值,更新當前根節點的值
        segTree[root].val = Max(segTree[2*root].val,segTree[2*root+1].val); /* 回溯時得到當前node節點的線段信息 */
    }
}

int query(int root, int nstart, int nend, int qstart, int qend)
{
    //查詢區間和當前節點區間沒有交集
    if(qstart > nend || qend < nstart)
        return -1;
    //當前節點區間包含在查詢區間內
    if(qstart <= nstart && qend >= nend)
        return segTree[root].val;
    //分別從左右子樹查詢,返回兩者查詢結果的較小值
    int mid = (nstart + nend) / 2;
    return Max(query(root*2, nstart, mid, qstart, qend),
               query(root*2+1, mid + 1, nend, qstart, qend));
 
}
void updateOne(int root, int nstart, int nend, int index, int addVal)
{
    if(nstart == nend)
    {
        if(index == nstart)//找到了相應的節點,更新之
            segTree[root].val = addVal;
        return;
    }
    int mid = (nstart + nend) / 2;
    if(index <= mid)//在左子樹中更新
        updateOne(root*2, nstart, mid, index, addVal);
    else updateOne(root*2+1, mid+1, nend, index, addVal);//在右子樹中更新
    //根據左右子樹的值回溯更新當前節點的值
    segTree[root].val = Max(segTree[root*2].val, segTree[root*2+1].val);
}

int main()
{
	int N,M;
	while(scanf("%d%d",&N,&M)!=EOF){
		for(int i=1;i<=N;i++){
			scanf("%d",&arr[i]);
		}
		getchar();	//6666666666666666666666666666666666666TL
		build(1,arr,1,N); 
		char ch;int a,b;
		while(M--){
			scanf("%c%d%d",&ch,&a,&b);
			getchar();				//6666666666666666666666666666666666
			if(ch==‘Q‘){
				printf("%d\n",query(1,1,N,a,b));
			}
			if(ch==‘U‘){
				updateOne(1,1,N,a,b);
			}
		}
	}
	return 0;
}

三、我的收獲

①當輸入與字符有關時,應考慮回車是否要處理;

②數組空間開的大小與TLE無關,只與MEL有關,也可能與RE有關。

線段樹——hdu1754I Hate It