poj 3264 Balanced Lineup(ST)
好几天没写博,还是没形成习惯啊,最近都在做线段树和树状数组方面的题,树状数组和线段树有一些相似的地方,有些题既能用线段树又能用树状数组,所以就都先用一下,不过还是不熟练。
那好吧,几天补齐前几天的博。
题意:给出N个数字和Q个询问,每个询问给出起始位置和终止位置,求出这段区间中最大数字和最小数字的差。
思路:RMQ的思想就是求一个区间中的最值,用ST求出区间最大值和最小值,然后求差。
代码:
View Code
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <iostream> #include <algorithm> #include <math.h> #define N 50004 using namespace std ; int mx[N][20] , mn[N][20] ; int maxs( int x , int y ) { return x > y ? x : y ; } int mins( int x , int y ) { return x < y ? x : y ; } void init( int n ) { int i , j , m ; m = floor( log( (double) n ) / log( 2.0)); for ( i = 1 ; i <= m ; i++ ) { for ( j = n ; j >= 1 ; j-- ) { if( j + ( 1<<( i-1) ) <= n ) mx[j][i] = maxs( mx[j][i-1] , mx[j+(1<<(i-1))][i-1] ); if ( j + ( 1<<( i-1)) <= n ) mn[j][i] = mins( mn[j][i-1] , mn[j+(1<<(i-1))][i-1] ); } } } int main() { int n , q , x , y , i ; while( scanf( "%d%d" , &n , &q ) != EOF ) { for ( i = 1 ; i <= n ; i++ ) { scanf( "%d" , &mx[i][0] ); mn[i][0] = mx[i][0] ; } init( n ); while( q-- ) { scanf( "%d%d" , &x , &y ); int m = floor( log( (double) ( y-x+1 ))/ log( 2.0 )); printf( "%d\n" , maxs( mx[x][m] , mx[y-(1<<m)+1][m] ) - mins( mn[x][m] , mn[y-(1<<m)+1][m] )); } } return 0 ; }