POJ 2823 线段树,RMQ

题意：给你n个数字，和一个长度m，输出所有长度为m的区间的最大值和最小值。

思路：直接线段树区间求最值就可以了。

或者可以用RMQ。

或者STL也可以。

贴个线段树吧。

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 0x3fffffff
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )
using namespace std;

#define N 1111111
int n , m ;
int a[N] ;
struct TT{
    int l , r , max , min ,num ;
}T[N * 3] ;
void build(int l , int r ,int x){
    T[x].l = l , T[x].r = r ;
    if(l == r){
        T[x].num = a[l] ;
        T[x].max = T[x].min = a[l] ;return ;
    }
    int mid = l + r >> 1 ;
    build(l , mid , LL(x)) ;
    build(mid + 1 , r , RR(x)) ;
    T[x].max = max(T[LL(x)].max , T[RR(x)].max) ;
    T[x].min = min(T[LL(x)].min , T[RR(x)].min) ;
}
int MAX , MIN ;
void query(int l , int r ,int x){
    if(T[x].l > r || T[x].r < l)return ;
    if(T[x].l == l && T[x].r == r){
        MAX = max(MAX , T[x].max) ;
        MIN = min(MIN , T[x].min) ;return ;
    }
    int mid = T[x].l + T[x].r >> 1 ;
    if(l > mid){
        query(l , r , RR(x)) ;
    }
    else if(r <= mid){
        query(l , r , LL(x)) ;
    }
    else {
        query(l , mid , LL(x)) ;
        query(mid + 1 , r , RR(x)) ;
    }
}
int b[N] , c[N] ;
int main() {
    scanf("%d%d",&n,&m) ;
    for (int i = 1 ; i <= n ; i ++ )scanf("%d",&a[i]) ;
    build(1 , n , 1) ;
    for (int i = 1 ; i <= n - m + 1 ; i ++ ){
        MAX = -inf , MIN = inf ;
        query(i , i + m - 1 ,1) ;
        b[i] = MIN ;
        c[i] = MAX ;
    }
    for (int i = 1 ; i <= n - m + 1 ; i ++)printf("%d ",b[i]) ;puts("") ;
    for (int i = 1 ; i <= n - m + 1 ; i ++)printf("%d ",c[i]) ;puts("") ;
    return 0;
}

再来个RMQ，需要注意的是，这道题一般的ST算法会MLE。但是考虑到他的区间长度是固定的，那么RMQ的第二维其实就不需要了，优化掉之后就OK了。

#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Max 2505
#define FI first
#define SE second
#define ll long long
#define PI acos(-1.0)
#define inf 0x3fffffff
#define LL(x) ( x << 1 )
#define bug puts("here")
#define PII pair<int,int>
#define RR(x) ( x << 1 | 1 )
#define mp(a,b) make_pair(a,b)
#define mem(a,b) memset(a,b,sizeof(a))
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )
using namespace std;

#define N 1001111
int n , m ;
int a[N] ;
int MAX[N] ,MIN[N] ;
//int LOG[N] ;
int LOG ;
void init() {
    for (int i = 1 ; i <= n ; i ++ ) {
        MAX[i] = MIN[i] = a[i] ;
    }
    for (int i = 1 ; i <= LOG ; i ++ ) {
        for (int j = 1 ; j <= n + 1 - (1 << i); j ++ ) {
            MAX[j] = max(MAX[j] ,MAX[j + (1 << (i - 1))] ) ;
            MIN[j] = min(MIN[j] ,MIN[j + (1 << (i - 1))] ) ;
        }
    }
}
int MIN_RMQ(int l ,int r) {
    int k = LOG ;
    return min(MIN[l] , MIN[r - (1 << k) + 1]) ;
}
int MAX_RMQ(int l ,int r) {
    int k = LOG ;
    return max(MAX[l] , MAX[r - (1 << k) + 1]) ;
}
//int b[N] , c[N] ;
int main() {
//    LOG[0] = -1 ;
//    for (int i = 1 ; i < N ; i ++ )LOG[i] = ((i & (i - 1)) == 0) ? LOG[i - 1] + 1 : LOG[i - 1] ;
    cin >> n >> m ;
    LOG = int(log(m * 1.0) / log(2 * 1.0)) ;
    for (int i = 1 ; i <= n ; i ++ )scanf("%d",&a[i]) ;
    init() ;
    for (int i = 1 ; i <= n - m + 1 ; i ++)printf("%d ",MIN_RMQ(i , i + m - 1)) ;puts("") ;
    for (int i = 1 ; i <= n - m + 1 ; i ++)printf("%d ",MAX_RMQ(i , i + m - 1)) ;puts("") ;
    return 0;
}