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

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

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;
}