線段樹——hdu1754I Hate It
阿新 • • 發佈:2017-08-04
cpp pri bsp clas rip 遞歸 有關 add 數組
這讓很多學生很反感。
不管你喜不喜歡,現在需要你做的是,就是按照老師的要求,寫一個程序,模擬老師的詢問。當然,老師有時候需要更新某位同學的成績。 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 二、解題思路
【評題】
一、題目回顧
題目鏈接: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