I Hate It 線段樹
阿新 • • 發佈:2018-11-12
很多學校流行一種比較的習慣。老師們很喜歡詢問,從某某到某某當中,分數最高的是多少。
這讓很多學生很反感。
不管你喜不喜歡,現在需要你做的是,就是按照老師的要求,寫一個程式,模擬老師的詢問。當然,老師有時候需要更新某位同學的成績。
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
Hint
Huge input,the C function scanf() will work better than cin
開始線段樹的學習了,然後只會一個sum的板子,然後各種不會,後來看懂了網上的程式碼,然後自己寫了一發超時
emmm,
就這樣了,這個是沒有lazy標記的,因為每次更新到點,那麼用了lazy反而更慢(超時就這麼來的)
Ac程式碼
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int maxn = 2e5+7; int a[maxn],n,m; char oper[20]; struct node{ int l,r,lazy; ll sum; }tree[maxn<<2]; void push_up(int x){ tree[x].sum = max(tree[x<<1].sum, tree[x<<1|1].sum); } void build(int x,int l,int r){ tree[x].l = l, tree[x].r = r; tree[x].lazy = tree[x].sum = 0; if(l == r){ tree[x].sum = a[l]; }else{ int mid = (l+r)>>1; build(x<<1,l,mid); build(x<<1|1,mid+1,r); push_up(x); } } void updata(int x,int pos,int val){ int L = tree[x].l, R = tree[x].r; if(L == R && R == pos){ tree[x].sum = val; }else{ int mid = (L+R)>>1; if(mid>=pos) updata(x<<1,pos,val); else updata(x<<1|1,pos,val); push_up(x); } } ll query(int x,int l,int r){ int L = tree[x].l, R = tree[x].r; if(l<=L && R<=r){ return tree[x].sum; }else{ int mid = (L+R)>>1; ll ans = 0; if(mid>=l) ans = max(ans, query(x<<1,l,r)); if(r>mid) ans = max(ans, query(x<<1|1,l,r)); push_up(x); return ans; } } int main(){ while(~scanf("%d%d",&n,&m)){ for(int i=1;i<=n;i++) scanf("%d",&a[i]); build(1,1,n); while(m--){ scanf("%s",oper); int l,r; if(oper[0]=='U'){ scanf("%d%d",&l,&r); //a[l] = r; updata(1,l,r); }else{ scanf("%d%d",&l,&r); printf("%d\n",query(1,l,r)); } } } return 0; }
超時一發
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+7;
int a[maxn],n,m;
char oper[20];
struct node{
int l,r,lazy;
ll sum;
void updata(int val){
sum += val;
lazy += val;
}
}tree[maxn<<2];
void push_up(int x){
tree[x].sum = max(tree[x<<1].sum, tree[x<<1|1].sum);
}
void push_down(int x){
ll lazy = tree[x].lazy;
if(lazy){
tree[x<<1].updata(lazy);
tree[x<<1|1].updata(lazy);
tree[x].lazy = 0;
}
}
void build(int x,int l,int r){
tree[x].l = l, tree[x].r = r;
tree[x].lazy = tree[x].sum = 0;
if(l == r){
tree[x].sum = a[l];
}else{
int mid = (l+r)>>1;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
push_up(x);
}
}
void updata(int x,int l,int r,int val){
int L = tree[x].l, R = tree[x].r;
if(l<=L && R<=r){
tree[x].updata(val);
}else{
int mid = (L+R)>>1;
push_down(x);
if(mid>=l) updata(x<<1,l,r,val);
if(r>mid) updata(x<<1|1,l,r,val);
push_up(x);
}
}
ll query(int x,int l,int r){
int L = tree[x].l, R = tree[x].r;
if(l<=L && R<=r){
return tree[x].sum;
}else{
int mid = (L+R)>>1;
ll ans = 0;
if(mid>=l) ans = max(ans, query(x<<1,l,r));
if(r>mid) ans = max(ans, query(x<<1|1,l,r));
push_up(x);
return ans;
}
}
int main(){
while(~scanf("%d%d",&n,&m)){
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,1,n);
while(m--){
scanf("%s",oper);
int l,r;
if(oper[0]=='U'){
scanf("%d%d",&l,&r);
updata(1,l,l,r-a[l]);
a[l] = r;
}else{
scanf("%d%d",&l,&r);
printf("%d\n",query(1,l,r));
}
}
}
return 0;
}