1. 程式人生 > 其它 >P1531 I Hate It

P1531 I Hate It

題面

很多學校流行一種比較的習慣。老師們很喜歡詢問,從某某到某某當中,分數最高的是多少。這讓很多學生很反感。

不管你喜不喜歡,現在需要你做的是,就是按照老師的要求,寫一個程式,模擬老師的詢問。當然,老師有時候需要更新某位同學的成績。

輸入格式

第一行,有兩個正整數 \(n\)\(m\)\(0<n \le 2\times 10^5,0<m<5000\)),分別代表學生的數目和操作的數目。學生 ID 編號分別從 \(1\) 編到 \(n\)。第二行包含 \(n\) 個整數,代表這 \(n\) 個學生的初始成績,其中第 \(i\) 個數代表 ID 為 \(i\) 的學生的成績。接下來有 \(m\)

行。每一行有一個字元 \(c\)(只取 QU),和兩個正整數 \(a\)\(b\)。當 \(c\)Q 的時候,表示這是一條詢問操作,它詢問 ID 從 \(a\)\(b\)(包括 \(a,b\)) 的學生當中,成績最高的是多少。當 \(c\)U 的時候,表示這是一條更新操作,如果當前 \(a\) 學生的成績低於 \(b\),則把 ID 為 \(a\) 的學生的成績更改為 \(b\),否則不改動。

輸出格式

對於每一次詢問操作,在一行裡面輸出最高成績。

思路

維護區間 \(\max\) 的線段樹,然後實現區間查詢+單點修改即可。

時間複雜度 \(O(m \log n)\)

程式碼

#include<bits/stdc++.h>
using namespace std;

const int SIZE = 2e5+5;
int t[SIZE<<2],a[SIZE];
int n,m;

void pushup(int i){
	t[i]=max(t[i<<1],t[i<<1|1]);
}

void build(int i,int l,int r){
	if(l==r){
		t[i]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(i<<1,l,mid);
	build(i<<1|1,mid+1,r);
	pushup(i);
}

int query(int ql,int qr,int l,int r,int i){
	if(ql<=l && r <= qr){
		return t[i];
	}
	int mid = (l+r)>>1;
	int result = 0;
	if(ql <= mid){
		result = max(result,query(ql,qr,l,mid,i<<1));
	}
	if(qr > mid){
		result = max(result,query(ql,qr,mid+1,r,i<<1|1));
	}
	return result;
}

void update(int p,int l,int r,int i,int v){
	if(l==r){
		if(t[i]<v){
			t[i]=v;
		}
		return;
	}
	int mid=(l+r)>>1;
	if(p <= mid){
		update(p,l,mid,i<<1,v);
	}
	if(p > mid){
		update(p,mid+1,r,i<<1|1,v);
	}
	pushup(i);
}

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,n);
	while(m--){
		char op;int a,b;
		cin>>op>>a>>b;
		if(op=='Q'){
			cout<<query(a,b,1,n,1)<<endl;
		}
		else{
			update(a,1,n,1,b);
		}
	}
	return 0;
}