伸展树——自顶向下

三种旋转

   当我们沿着树向下搜索某个节点X的时候,我们将搜索路径上的节点及其子树移走。我们构建两棵临时的树──左树和右树。

没有被移走的节点构成的树称作中树。在伸展操作的过程中:

1、当前节点X是中树的根。
2、左树L保存小于X的节点。
3、右树R保存大于X的节点。
开始时候,X是树T的根,左右树L和R都是空的。
1、zig:
 
                               
    如上图,在搜索到X的时候,所查找的节点比X小,将Y旋转到中树的树根。旋转之后,X及其右子树被移动到右树上。很显然,右树上的节点都大于所要查找的节点。注意X被放置在右树的最小的位置,也就是X及其子树比原先的右树中所有的节点都要小。这是由于越是在路径前面被移动到右树的节点,其值越大。
2、zig-zig:
 
                               
    在这种情况下,所查找的节点在Z的子树中,也就是,所查找的节点比X和Y都小。所以要将X,Y及其右子树都移动到右树中。首先是Y绕X右旋,然后Z绕Y右旋,最后将Z的右子树(此时Z的右子节点为Y)移动到右树中。注意右树中挂载点的位置。
3、zig-zag:
 
                           
    在这种情况中,首先将Y右旋到根。这和Zig的情况是一样的。然后变成上图右边所示的形状。接着,对Z进行左旋,将Y及其左子树移动到左树上。这样,这种情况就被分成了两个Zig情况。这样,在编程的时候就会简化,但是操作的数目增加(相当于两次Zig情况)。
    最后,在查找到节点后,将三棵树合并。如图:
 
                               

    将中树的左右子树分别连接到左树的右子树和右树的左子树上。将左右树作为X的左右子树。重新最成了一所查找的节点为根的树。

   

   

  右连接:将当前根及其右子树连接到右树上。左子结点作为新根。
  左连接:将当前根及其左子树连接到左树上。右子结点作为新根。

 越是在路径前面被移动到右树的节点,其值越大;越是在路径前面移动到左树的节点,其值越小。

 这很显然,右连接,将当前根以及右子树全部连接到右树,即把整课树中值大的一部分移走(大于等于当前根),

 后面,在进行右连接,其值肯定小于之前的,所以,要加在右树的左边。


伸展树伪代码

 右连接:将当前根及其右子树连接到右树上。左子结点作为新根。
 左连接:将当前根及其左子树连接到左树上。右子结点作为新根。
  T : 当前的根节点。
  Function Top-Down-Splay
     Do
          If X 小于 T Then
               If X 等于 T 的左子结点 Then           //zig
                 右连接
               ElseIf X 小于 T 的左子结点 Then       //zig-zig
                 T的左子节点绕T右旋
                 右连接
               Else X大于 T 的左子结点 Then          //zig-zag
                 右连接
                 左连接
               EndIf    
           ElseIf X大于 T Then
               IF X 等于 T 的右子结点 Then
                 左连接
               ElseIf X 大于 T 的右子结点 Then
                 T的右子节点绕T左旋
                 左连接
               Else X小于 T 的右子结点 Then
                 左连接
                 右连接
               EndIf
          EndIf
     While  !(找到 X或遇到空节点)
      组合左中右树
  EndFunction

 

zig-zag这种情形,可以先按zig处理。第二次循环在进行一次处理。简化代码如下:

  Function Top-Down-Splay
      Do
          If X 小于 T Then
               If X 小于 T 的左孩子 Then
                  T的左子节点绕T右旋
               EndIf    
               右连接
              Else If X大于 T Then
                If X 大于 T 的右孩子 Then
                    T的右子节点绕T左旋
                EndIf
                左连接
              EndIf
      While  !(找到 X或遇到空节点)
      组合左中右树
  EndFuntion

例子

下面是一个查找节点19的例子:
    在例子中,树中并没有节点19,最后,距离节点最近的节点18被旋转到了根作为新的根。节点20也是距离节点19最近的节点,但是节点20没有成为新根,这和节点20在原来树中的位置有关系。
 
  如果查找的元素不在树中,则寻找过程中出现空的上一个节点即为根节点。

CPP代码

 .h

struct SplayNode
{
	int element;
	SplayNode *left;
	SplayNode *right;
};

class SplayTree
{
	public:
		SplayTree();
		~SplayTree();
		
		void Insert(int data);
		void Remove(int data);
		SplayNode *Find(int data);

		
	private:
		SplayNode *_tree;
		
		SplayNode *_Insert(SplayNode *T, int item);
		SplayNode *_Remove(SplayNode *T, int item);
		SplayNode *_Splay(SplayNode *X, int Item);
		
		SplayNode *_SingleToRight(SplayNode *K2);  
                SplayNode *_SingleToLeft(SplayNode *K2); 
  
                void  _MakeEmpty(SplayNode *root);
			
};

.cpp

SplayTree::SplayTree()
{
	_tree = NULL;
}


SplayTree::~SplayTree()
{
	_MakeEmpty(_tree);
}

void SplayTree::Insert(int data)
{
	_tree = _Insert(_tree, data);
}

void SplayTree::Remove(int data)
{
	_tree = _Remove(_tree, data);
}

SplayNode *SplayTree::_SingleToRight(SplayNode *K2)  
{  
    SplayNode *K1 = K2->left;  
    K2->left = K1->right;  
    K1->right = K2;  
        
    return K1;  
}  
  
          
SplayNode *SplayTree::_SingleToLeft(SplayNode *K2)  
{  
    SplayNode *K1 = K2->right;  
    K2->right = K1->left;  
    K1->left  = K2;  
          
    return K1;  
} 
		
SplayNode *SplayTree::_Splay(SplayNode *X, int item)
{
	static struct SplayNode Header;
	SplayNode *leftTreeMax, *rightTreeMin;
	
	Header.left = Header.right = NULL;
	leftTreeMax = rightTreeMin = &Header;
	
	while( X != NULL && item != X->element)
	{

		if(item < X->element)
		{
			if(X->left == NULL)
				break;
			if(item < X->left->element)
			{
				X = _SingleToRight(X); 
			} 
			if( X->left == NULL)
				break;
                        //右连接
                        rightTreeMin->left = X;
			rightTreeMin       = X;
			X                  = X->left;
		}
		else
		{
			if(X->right == NULL)
				break;
			if(item > X->right->element)
			{
				X = _SingleToLeft(X);
			}
			if(X->right == NULL)
				break;
                        //左连接
                        leftTreeMax->right = X;
			leftTreeMax        = X;
			X                  = X->right;
		}
	}
	leftTreeMax->right = X->left;
	rightTreeMin->left = X->right;
	X->left = Header.right;
	X->right = Header.left;
	
	return X;
}


SplayNode *SplayTree::_Insert(SplayNode *T, int item)
{
	static SplayNode* newNode = NULL;
	
	if(newNode == NULL)
	{
		newNode = new SplayNode;
	}
	newNode->element = item;
	
	if(T == NULL)
	{
		newNode->left = newNode->right = NULL;
		T = newNode;
	}
	else
	{
		T = _Splay(T, item);
		if(item < T->element)
		{
			newNode->left = T->left;
			newNode->right = T;
			T->left = NULL;
			T = newNode;
		}
		else if(T->element < item)
		{
			newNode->right = T->right;
			newNode->left = T;
			T->right = NULL;
			T = newNode;
		}
		else
		{
			delete newNode;
		}
	}
	newNode = NULL;
	return T;
}

SplayNode *SplayTree::_Remove(SplayNode *T, int item)
{
	SplayNode *newTree;
	
	if(T != NULL)
	{
		T = _Splay(T, item);
		if(item == T->element)
		{
			if(T->left == NULL)
			{
				newTree = T->right;
			}
			else
			{
				newTree = T->left;
				newTree = _Splay(newTree, item);
				newTree->right = T->right;
			}
			delete T;
			T = newTree;
		}
	}
	return T;
}

void SplayTree::_MakeEmpty(SplayNode *T)  
{  
    if(T != NULL)  
    {  
        _MakeEmpty(T->left);  
        _MakeEmpty(T->right);  
        delete T;  
    }  
}


SplayNode *SplayTree::Find(int data)
{
	_tree = _Splay(_tree, data);
	if(_tree->element == data)
		return _tree;
	else
		return NULL;
}

参考

参考:http://www.cnblogs.com/kernel_hcy/archive/2010/03/17/1688360.html

你可能感兴趣的:(数据结构,数据结构)