POJ 2823 Sliding Window 線段樹求解

題目鏈接:http://poj.org/problem?id=2823

題意:就是求區間最小值。

此題,顯然可以用線段樹過,但是,知道此題是學習滑動窗口時。此題用滑動窗口做的話比較省代碼,內存和時間。但是要是真正到了賽場上,可能真的不太能想起用滑動窗口,而且對單調隊列也不是很熟悉。寫一下線段樹也還是不錯的。


代碼:

#include<iostream>
#include<cstdio>
#include<algorithm>
#define maxn 1000010

using namespace std;

struct node{
    int l,r,Min,Max;
}T[maxn*3];

int A[maxn],B[maxn],C[maxn],N,K;

void build(int L,int R,int o){
    T[o].l=L;T[o].r=R;//居然把這裏給忘了
    if(L==R){
        T[o].Min=T[o].Max=A[L];return;
    }
    int tx,ttx,ty,tty;
    build(L,(L+R)/2,o*2);
    build((L+R)/2+1,R,o*2+1);
    T[o].Min=min(T[o*2].Min,T[o*2+1].Min);
    T[o].Max=max(T[o*2].Max,T[o*2+1].Max);
}

void query(int L,int R,int &x,int &y,int o){
    if(T[o].l>=L && T[o].r<=R){
        x=T[o].Min;y=T[o].Max;
        return;
    }
    if(T[o].l==T[o].r) return;//居然把這裏也給忘了
    if(L>=T[o*2+1].l) query(L,R,x,y,o*2+1);
    else if(R<=T[o*2].r) query(L,R,x,y,o*2);
    else{
        int tx,ty,ttx,tty;
        query(L,T[o*2].r,tx,ty,o*2);
        query(T[o*2+1].l,R,ttx,tty,o*2+1);
        x=min(tx,ttx);y=max(ty,tty);
    }
}

int main(){
    cin>>N>>K;
    for(int i=1;i<=N;i++) scanf("%d",A+i);
    int a,b,cnt=0;
    build(1,N,1);
    for(int i=K;i<=N;i++){
        query(i-K+1,i,a,b,1);
        B[cnt]=a;C[cnt]=b;cnt++;
    }
    for(int i=0;i<cnt;i++)
        printf("%d%c",B[i],i==cnt-1? '\n':' ');
    for(int i=0;i<cnt;i++)
        printf("%d%c",C[i],i==cnt-1? '\n':' ');
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章