【POJ】3580 SuperMemo 【splay】

最新推荐文章于 2018-03-04 20:35:50 发布

poursoul

最新推荐文章于 2018-03-04 20:35:50 发布

阅读量1k

点赞数

分类专栏：伸展树【splay】文章标签： poj

本文链接：https://blog.csdn.net/u013368721/article/details/38588037

版权

伸展树【splay】专栏收录该内容

8 篇文章 0 订阅

订阅专栏

传送门：【POJ】3580 SuperMemo

题目分析：这道题又——调了一天。。。。真是伤啊。。。。

题目不难，都是splay的基本操作，但是由于我代码能力太差各种写搓。。。尤其今天是败在了删除一个节点上，写了数据生成器才发现remove函数彻底写错了。。。。T u T。。一天就这么过去了。。。

这题没给数据范围，其实int就够了，最大值别设太小就行，一般0x3f3f3f3f够了。还有就是其中将一个区间循环的操作需要注意，不仅循环次数很大需要取模，而且还可能给你来负的！

大概讲解一下6个操作的解决方法：

1.ADD 旋转出区间[L,R]（即l-1为根结点，r+1为根结点的右节点，此时区间[L,R]，为根结点的右节点的子树），然后对子树的根打上标记。和线段树一样，lazy标记呗。

2.REVERSE 同样旋转出区间[L,R]，对[L,R]子树的根的标记取反。

3.REVOLVE 其实就是得到一个取模后的T，然后将[L,R-T]插入到已经删除了这个区间的位置L-1+T的右端，即将L-1+T旋转到根，再将L+T旋转为根的右节点，然后将区间[L,R-T]变为根结点的右节点的左节点即可。

4.INSERT 新建一个节点，然后用类似操作3的方法插入。

5.DELETE 将要删除的数所在的位置pos作为一个区间[pos,pos]旋转出来，然后像切树一样切掉就行了。

6.MIN 只要在pushup以及pushdown维护一下子树的最小值，然后用旋转区间的方法旋转出区间[L,R]，直接返回这个区间子树的根结点的信息即可。

代码如下：

/*
	本算法需要两个虚拟节点：0、n+1
	算法基本函数功能：
	clear ()                                        初始化 
	newNode ( int v , SplayNode* f )                新建节点并赋值v以及令其父亲为f
	init ( int n , int num[] )                      初始化
	rotate ( SplayNode* o , int d )                 左右旋，d=0表示左旋，d=1表示右旋
	splay ( SplayNode* o , SplayNode* f )           伸展o直到o的父节点为f为止
	select ( int k , SplayNode* f )                 选择第k个元素并将其旋转到其父节点为f为止（不计虚拟节点）
	SplayNode* get ( int l , int r )                返回区间[l,r]为子树的根:即将l-1旋转到根，将r+1旋转为l-1的右节点
	void split ( int l , int r , SplayNode* &o )    将区间[l,r]剪切到o
	void merge ( int pos , SplayNode* o )           将o插入到pos后端：即将pos旋转到根，pos+1旋转为pos的右节点
	void cut ( int l , int r , int pos )            剪出区间[l,r]并插入到pos的后端
	void reverse ( int l , int r )                  翻转区间[l,r]
	void revolve ( int l , int r , int T )          剪出区间[l,R-T]，然后在删除了该区间后的位置l-1+T后端插入该区间
	void getMin ( int l , int r )                   返回区间[l,r]最小值
	void add ( int l , int r , int v )              对区间[l,r]内的所有数加上v
	void insert ( int pos , int v )                 在pos的后端插入值为v的新节点
	void remove ( int pos )                         删除位置为pos的节点
	void print ( SplayNode* o )                     输出信息
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

#define REP( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define FOR( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define REV( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define CLR( a , x ) memset ( a , x , sizeof a )

const int MAXN = 233333 ;
const int INF = 0x3f3f3f3f ;

struct SplayNode {
	SplayNode *c[2] , *f ;
	int s ;
	int v ;
	bool rev ;
	int addv ;
	int minv ;
	
	void add ( int val ) {
		v += val ;
		addv += val ;
		minv = min ( v , minv ) ;
	}
	
	void pushup () {
		s = c[0] -> s + c[1] -> s + 1 ;
		minv = min ( c[0] -> minv , c[1] -> minv ) ;
		minv = min ( minv , v ) ;
	}
	
	void pushdown () {
		if ( rev ) {
			swap ( c[0] , c[1] ) ;
			c[0] -> rev ^= 1 ;
			c[1] -> rev ^= 1 ;
			rev = 0 ;
		}
		if ( addv ) {
			if ( ~c[0] -> v ) {
				c[0] -> v += addv ;
				c[0] -> addv += addv ;
				c[0] -> minv += addv ;
			}
			if ( ~c[1] -> v ) {
				c[1] -> v += addv ;
				c[1] -> addv += addv ;
				c[1] -> minv += addv ;
			}
			addv = 0 ;
		}
	}
} Tnull , *null = &Tnull ;

struct Splay {
	SplayNode node[MAXN] , *cur ;
	SplayNode *root ;
	
	void clear () {
		null -> s = 0 ;
		null -> v = -1 ;
		null -> minv = INF ;
		cur = node ;
		root = null ;
	}
	
	SplayNode* newNode ( int v , SplayNode* f ) {
		cur -> v = v ;
		cur -> s = 1 ;
		cur -> rev = 0 ;
		cur -> addv = 0 ;
		cur -> minv = v ;
		cur -> c[0] = cur -> c[1] = null ;
		cur -> f = f ;
		return cur ++ ;
	}
	
	void rotate ( SplayNode* o , bool d ) {
		SplayNode* p = o -> f ;
		p -> pushdown () ;
		o -> pushdown () ;
		p -> c[!d] = o -> c[d] ;
		if ( o -> c[d] != null ) o -> c[d] -> f = p ;
		o -> f = p -> f ;
		if ( p -> f != null ) {
			if ( p == p -> f -> c[0] ) p -> f -> c[0] = o ;
			else p -> f -> c[1] = o ;
		}
		o -> c[d] = p ;
		p -> f = o ;
		p -> pushup () ;
		if ( root == p ) root = o ;
	}
	
	void splay ( SplayNode* o , SplayNode* f ) {
		while ( o -> f != f ) {
			SplayNode* p = o -> f ;
			if ( p -> f == f ) {
				if ( o == p -> c[0] ) rotate ( o , 1 ) ;
				else				  rotate ( o , 0 ) ;
			} else {
				if ( p == p -> f -> c[0] ) {
					if ( o == p -> c[0] ) rotate ( p , 1 ) , rotate ( o , 1 ) ;
					else				  rotate ( o , 0 ) , rotate ( o , 1 ) ;
				} else {
					if ( o == p -> c[1] ) rotate ( p , 0 ) , rotate ( o , 0 ) ;
					else				  rotate ( o , 1 ) , rotate ( o , 0 ) ;
				}
			}
		}
		o -> pushup () ;
	}
	
	void select ( int k , SplayNode* f ) {
		SplayNode* o = root ;
		++ k ;
		while ( o != null ) {
			o -> pushdown () ;
			int s = o -> c[0] -> s ;
			if ( k == s + 1 ) break ;
			if ( k <= s ) o = o -> c[0] ;
			else {
				k -= s + 1 ;
				o = o -> c[1] ;
			}
		}
		splay ( o , f ) ;
	}
	
	SplayNode* get ( int l , int r ) {
		select ( l - 1 , null ) ;
		select ( r + 1 , root ) ;
		return root -> c[1] -> c[0] ;
	}
	
	void split ( int l , int r , SplayNode* &o ) {
		o = get ( l , r ) ;
		o -> f = null ;
		root -> c[1] -> c[0] = null ;
		root -> c[1] -> pushup () ;
		root -> pushup () ;
	}
	
	void merge ( int pos , SplayNode* o ) {
		get ( pos + 1 , pos ) ;
		o -> f = root -> c[1] ;
		root -> c[1] -> c[0] = o ;
		root -> c[1] -> pushup () ;
		root -> pushup () ;
		splay ( o , null ) ;
	}
	
	void cut ( int l , int r , int pos ) {
		SplayNode* o ;
		split ( l , r , o ) ;
		merge ( pos , o ) ;
	}
	
	void reverse ( int l , int r ) {
		SplayNode* o = get ( l , r ) ;
		o -> rev ^= 1 ;
		splay ( o , null ) ;
	}
	
	void revolve ( int l , int r , int T ) {
		cut ( l , r - T , l - 1 + T ) ;
	}
	
	void add ( int l , int r , int v ) {
		SplayNode* o = get ( l , r ) ;
		o -> add ( v ) ;
		splay ( o , null ) ;
	}
	
	void insert ( int pos , int v ) {
		get ( pos + 1 , pos ) ;
		SplayNode* o = newNode ( v , root -> c[1] ) ;
		root -> c[1] -> c[0] = o ;
		root -> c[1] -> pushup () ;
		root -> pushup () ;
	}
	
	void remove ( int pos ) {
		get ( pos , pos ) ;
		root -> c[1] -> c[0] = null ;
		root -> c[1] -> pushup () ;
		root -> pushup () ;
	}
	
	int getMin ( int l , int r ) {
		SplayNode* o = get ( l , r ) ;
		return o -> minv ;
	}
	
	void init ( int n , int num[] ) {
		clear () ;
		FOR ( i , 0 , n + 1 ) {
			SplayNode* o = newNode ( num[i] , null ) ;
			o -> c[0] = root ;
			root -> f = o ;
			root = o ;
			root -> pushup () ;
		}
	}
	
	void print ( SplayNode* o ) {
		if ( o -> c[0] != null ) print ( o -> c[0] ) ;
		printf ( "左儿子v:%d----自己v:%d----右儿子v:%d\n" , o -> c[0] -> v , o -> v , o -> c[1] -> v ) ;
		if ( o -> c[1] != null ) print ( o -> c[1] ) ;
	}
} T ;

int n , p ;
int num[MAXN] ;
char s[10] ;
int x , y , v ;

void solve () {
	num[0] = num[n + 1] = INF ;
	FOR ( i , 1 , n ) scanf ( "%d" , &num[i] ) ;
	T.init ( n , num ) ;
	scanf ( "%d" , &p ) ;
	while ( p -- ) {
		scanf ( "%s" , s ) ;
		if ( s[0] == 'A' ) {
			scanf ( "%d%d%d" , &x , &y , &v ) ;
			T.add ( x , y , v ) ;
		} else if ( s[0] == 'R' && s[3] == 'E' ) {
			scanf ( "%d%d" , &x , &y ) ;
			T.reverse ( x , y ) ;
			
		} else if ( s[0] == 'R' && s[3] == 'O' ) {
			scanf ( "%d%d%d" , &x , &y , &v ) ;
			v %= y - x + 1 ;
			if ( v < 0 ) v += y - x + 1 ;
			T.revolve ( x , y , v ) ;
		} else if ( s[0] == 'I' ) {
			scanf ( "%d%d" , &x , &v ) ;
			T.insert ( x , v ) ;
		} else if ( s[0] == 'D' ) {
			scanf ( "%d" , &x ) ;
			T.remove ( x ) ;
		} else {
			scanf ( "%d%d" , &x , &y ) ;
			printf ( "%d\n" , T.getMin ( x , y ) ) ;
		}
	}
}

int main () {
	while ( ~scanf ( "%d" , &n ) ) solve () ;
	return 0 ;
}

为方便调试，另附小数据生成器：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stdlib.h>
#include <time.h>
using namespace std ;

#define REP( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define FOR( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define REV( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define CLR( a , x ) memset ( a , x , sizeof a )

const int R = 100 ;

int main () {
	srand ( time ( NULL ) ) ;
	int n , x , y , q , ch ;
	freopen ( "poj3580.txt" , "w" , stdout ) ;
	REP ( Q , 0 , 10 ) {
		n = rand () % 100 + 1 ;
		printf ( "%d\n" , n ) ;
		REP ( i , 0 , n ) printf ( "%d " , rand () % R + 1 ) ;
		printf ( "\n" ) ;
		q = rand () % 100 + 1 ;
		printf ( "%d\n" , q ) ;
		while ( q -- ) {
			ch = rand () % 6 + 1 ;
			if ( ch == 1 ) {
				x = rand () % n + 1 ;
				while ( ( y = rand () % n + 1 ) < x ) ;
				printf ( "ADD %d %d %d\n" , x , y , rand () % R + 1 ) ;
			} else if ( ch == 2 ) {
				x = rand () % n + 1 ;
				while ( ( y = rand () % n + 1 ) < x ) ;
				printf ( "REVERSE %d %d\n" , x , y ) ;
			} else if ( ch == 3 ) {
				x = rand () % n + 1 ;
				while ( ( y = rand () % n + 1 ) < x ) ;
				printf ( "REVOLVE %d %d %d\n" , x , y , rand () % R + 1 ) ;
			} else if ( ch == 4 ) {
				printf ( "INSERT %d %d\n" , rand () % n + 1 , rand () % R + 1 ) ;
				++ n ;
			} else if ( ch == 5 ) {
				printf ( "DELETE %d\n" , rand () % n + 1 ) ;
				-- n ;
			} else {
				x = rand () % n + 1 ;
				while ( ( y = rand () % n + 1 ) < x ) ;
				printf ( "MIN %d %d\n" , x , y ) ;
			}
		}
	}
	return 0 ;
}