Chinaunix首页 | 论坛 | 博客
  • 博客访问: 126206
  • 博文数量: 36
  • 博客积分: 94
  • 博客等级: 民兵
  • 技术积分: 200
  • 用 户 组: 普通用户
  • 注册时间: 2009-02-23 17:34
文章分类
文章存档

2015年(1)

2013年(7)

2012年(3)

2011年(25)

分类:

2011-08-26 14:35:46

原文地址:treap的C实现 作者:Bean_lee

    谨以此文,向byvoid大牛致敬 
    treap,在平衡二叉搜索树中是一种相对简单的数据结构,它的本质是tree +heap。
    它即有二叉树的特性,即节点的左孩子的关键值(key)小于节点的关键值,节点右孩子的关键值大于节点的关键值。 我们前文讲二叉搜索树的随机化策略的时候提到过,二叉搜索树可能退化成链表,降低了搜索性能。所以需要进行各种处理来防止这种退化的产生。在我的博文 随机二叉搜索树的C实现 中提到了一种解决的办法,即通过随机化的策略,使每个节点成为根的概率相等。
      treap维护了一个优先级的字段,也就是heap部分。对于最小堆而言,节点的左右孩子的优先级(priority)要高于节点本身的优先级。
      对于treap的插入而言,分两步走:
      1 和普通的二叉搜索树一样,根据key的值,插入到二叉树
      2通过旋转调整。新插入节点 可能破坏了heap的性质,即新节点可能优先级低于其父亲节点。需要调整。

     如下图所示 (图来自byvoid大牛的文章,如有侵权,请通知我立即删除)
     第一步 插入

第二步 根据优先级字段调整 


不满足heap的要求,继续调整(4节点的优先级15 低于5节点的20,不符合最小堆的要求)
     


     还是没解释清楚,为什么加上 优先级字段,维护了heap的性质,就保证了二叉树的平衡。秘密在于,优先级字段的值是随机值。换句话说,就是我们随机指定优先级字段。这种随机性来保证二叉树的平衡。

    讲完插入,继续解释删除
    情况一,该节点为叶节点或链节点,则该节点是可以直接删除的节点。若该节点有非空子节点,用非空子节点代替该节点的,否则用空节点代替该节点,然后删除该节点。
     情况二,该节点有两个非空子节点。我们的策略是通过旋转,使该节点变为可以直接删除的节点。如果该节点的左子节点的优先级小于右子节点的优先级,右旋该节点,使该节点降为右子树的根节点,然后访问右子树的根节点,继续讨论;反之,左旋该节点,使该节点降为左子树的根节点,然后访问左子树的根节点,继续讨论,知道变成可以直接删除的节点。

情况1 

   

  情况2

-------------------------------------treap.h-------------------------------------------------------------
#ifndef _TREAP_H
#define _TREAP_H

typedef int  key_type ;
typedef int  ;

typedef struct _treap_node
{
struct _treap_node *left;
struct _treap_node *right;
key_type key;
prio_type priority;
int data;
}treap_node;

typedef struct _treap
{
treap_node* root;
}treap;

treap_node* treap_insert(treap_node* root,int key,int data);
treap_node* treap_delete(treap_node* root,int key);

#endif
----------------------------------------treap.c-----------------------------------------------------
#include
#include"treap.h"
#include
#include

treap_node*  treap_left_rotate(treap_node *node)
{
assert(node->right != NULL);
treap_node* right = node->right;
node->right = right->left;
right->left = node;
return  right;
}

treap_node* treap_right_rotate(treap_node* node)
{
assert(node->left !=NULL)

treap_node* left = node->left;
node->left = left->right;
left->right->node;

return left;
}


treap_node* treap_allocnode(key_type key,int data)
{
treap_node* node = NULL;
node = (treap_node*)malloc(sizeof(treap_node));
if(node == NULL)
return NULL;
else
{
node->left = node->right = NULL;
node->key = key;
node->data = data;
node->priority = rand()*rand();
return node;
}
}

void treap_freenode(treap_node* node)
{
    if(node)
      free(node);
}

treap_node* treap_insert(treap_node* root,int key,int data)
{
if(root == NULL)
{
treap_node *new_node = treap_allocnode(key,data);
return new_node;
}

if(root->key > key)
{
root->left = treap_insert(root->left,key,data);
if(root->left->priority < root->priority)
return treap_right_rotate(root);
else
return root;
}
if(root->key < key)
{
root->right = treap_insert(root->right,key,data);
if(root->right->priority < root->priority)
return treap_left_rotate(root);
else
return root;
}
}

treap_node* treap_delete(treap_node* root,int key)
{
    
    if(root == NULL)
    {
      return NULL;/*not find a element with key equal to key*/
    }
    if(root->key == key)
    {
        if(root->left == NULL || root->right == NULL)
        {
            treap_node* delnode =root;
            if(root->left == NULL)
                root = root->right;
            else
                root = root->left;
           
            treap_freenode(delnode);
            return root;            
        }
        else
        {
            if(root->left->priority < root->right->priority)
            {
                treap_node* retnode = root->left;
                treap_right_rotate(root);
                retnode->right = treap_delete(root);
                return retnode;
            }
            else
            {
                treap_node* retnode = root->right;
                treap_left_rotate(root);
                retnode->left = treap_delete(root);
                return retnode;
            }
       }
      
    }
    else
    {
      if(root->key < key)
       root->right = treap_delete(root->right,key);
       
       else
        root->left = treap_delete(root->left,key);
        return root;
    }
    
}

------------------------------treap_test.c-------------------------------------------------
#include
#include
#include

#define NODE_NUM 10
#define random(x) (rand()%x)
void test_treap()
{
  int i;
  int array_random[NODE_NUM] ;
  for(i = 0;i
     array_random[i] = rand(10000);
  
  treap my_treap;
  my_treap.root = NULL;
  
  for(i = 0;i
  {
    my_treap.root = treap_insert(my_treap.root,array_random[i],array_random[i]);
  }
  
}


int main()
{
  test_treap();
  return 0;
}
   
    我相信总有一篇文章能清楚明白的解释我想了解的知识。对于treap这种数据结构,这篇文章无疑是:
byvoid大牛的文章《随机平衡二叉查找树 Treap 的分析与应用》,强烈推荐byvoid的这篇文章。看了这篇文章,基本上就不需要看听我在此JJYY了。
    
参考文献:
1 byvoid大牛的文章《随机平衡二叉查找树 Treap 的分析与应用》
2 维基百科 


阅读(1415) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~