hdu 1754 I Hate It (splay tree伸展树)

hdu 1754 I Hate It

其实我只是来存一下我的splay模板的。。请大牛们多多指教

 

#include<stdio.h>

#include<string.h>

#include<algorithm>

using namespace std ;



const int maxn = 222222 ;



int son[2][maxn] , col[maxn] , fa[maxn] , size[maxn] , val[maxn] ;

int tot ;

int num[maxn] ;



void new_node () {

	size[tot] = 1 ;

	fa[tot] = son[0][tot] = son[1][tot] = -1 ;

	val[tot] = col[tot] = 0 ;

	tot ++ ;

}



void update ( int rt ) {

	size[rt] = 1 ;

	col[rt] = val[rt] ;

	if ( son[0][rt] != -1 ) 

		size[rt] += size[son[0][rt]] , col[rt] = max ( col[rt] , col[son[0][rt]] ) ;

	if ( son[1][rt] != -1 )

		size[rt] += size[son[1][rt]] , col[rt] = max ( col[rt] , col[son[1][rt]] ) ;

}



int build ( int l , int r ) {

	if ( l > r ) return -1 ;

	int mid = ( l + r ) >> 1 ;

	new_node () ;

	int temp = tot - 1 ;

	val[temp] = num[mid] ;

	son[0][temp] = build ( l , mid - 1 ) ;  

	if ( son[0][temp] != -1 ) fa[son[0][temp]] = temp ;

	son[1][temp] = build ( mid + 1 , r ) ;

	if ( son[1][temp] != -1 ) fa[son[1][temp]] = temp ;

	update ( temp ) ;

	return temp ;

}



void rot ( int rt , int c ) {

	int y = fa[rt] , z = fa[y] ;

	son[!c][y] = son[c][rt] ;

	if ( son[c][rt] != -1 ) fa[son[c][rt]] = y ;

	fa[rt] = z ;

	if ( z != -1 ) {

		if ( y == son[0][z] ) son[0][z] = rt ;

		else son[1][z] = rt ;

	}

	son[c][rt] = y , fa[y] = rt ;

	update ( y ) ;

}



void splay ( int rt , int to ) {

	while ( fa[rt] != to ) {

		if ( fa[fa[rt]] == to ) {

			if ( rt == son[0][fa[rt]] ) rot ( rt , 1 ) ;

			else rot ( rt , 0 ) ;

		}

		else {

			int y = fa[rt] , z = fa[y] ;

			if ( rt == son[0][y] ) {

				if ( y == son[0][z] ) rot ( y , 1 ) , rot ( rt , 1 ) ;

				else rot ( rt , 1 ) , rot ( rt , 0 ) ;

			}

			else {

				if ( y == son[1][z] ) rot ( y , 0 ) , rot ( rt , 0 ) ;

				else rot ( rt , 0 ) , rot ( rt , 1 ) ;

			}

		}

	}

	update ( rt ) ;

	/*为什么只要在最后更新rt节点呢？因为在旋转的过程中，fa[rt]总是旋到rt下面的

	所以在旋转的时候，我们不断的更新fa[rt]，而不需要一直更新rt，只要在splay之后更新rt就好了

	*/

}



int find ( int rt , int key ) {

	int cnt = 0 ;

	if ( son[0][rt] != -1 ) cnt = size[son[0][rt]] ;

	if ( cnt + 1 == key ) return rt ;

	if ( cnt >= key ) return find ( son[0][rt] , key ) ;

	return find ( son[1][rt] , key - cnt - 1 ) ;

}



int main () {

	int n , m , i , j , k , a , b ;

	char op[11] ;

	while ( scanf ( "%d%d" , &n , &m ) != EOF ) {

		for ( i = 1 ; i <= n ; i ++ ) scanf ( "%d" , &num[i] ) ;

		tot = 0 ;

		int root = build ( 0 , n + 1 ) , temp ;

		while ( m -- ) {

			scanf ( "%s%d%d" , op , &a , &b ) ;

			a ++ , b ++ ;//因为建树的时候，把0也建进去了，但实际上0是不在原数组中的，所有往后推移一位

			if ( op[0] == 'U' ) {

				temp = find ( root , a ) ;

				splay ( temp , -1 ) ;

				root = temp ;

				val[root] = b - 1 ;

			//	update ( root ) ;

			//这里为什么不用更新rt呢？因为对于size的值是不会变的，而val值已经更新了，而对于col值，下次是splay时，肯定会更新掉的

			}

			else {

				temp = find ( root , a - 1 ) ;

				splay ( temp , -1 ) ;

				root = temp ;

				temp = find ( root , b + 1 ) ;

				splay ( temp , root ) ;

				printf ( "%d\n" , col[son[0][temp]] ) ;

			}

		}

	}

}