码迷,mamicode.com
首页 > 其他好文 > 详细

BST - treap

时间:2017-11-06 22:47:05      阅读:139      评论:0      收藏:0      [点我收藏+]

标签:erase   treap   des   span   lib   name   turn   ras   printf   

#include<cstdio>
#include<cstdlib>
using namespace std;
struct treapNode {
    int val,ra;
    treapNode *left,*right;
};

void rightRoate(treapNode *&root) {
    treapNode *right = root->right;
    root->right = right->left;
    right->left = root;
    root = right;
}

void leftRoate(treapNode *&root) {
    treapNode *left = root->left;
    root->left = left->right;
    left->right = root;
    root = left;
}

void insert(treapNode *&root,int val) {
    if(!root) {
        root = new treapNode;
        root->val = val;
        root->ra = rand()*rand();
        root->left = root->right = NULL;
        return;
    }
    if(val <= root->val) {
        insert(root->left,val);
        if(root->ra > root->left->ra) {
            leftRoate(root);
        }
    }
    else {
        insert(root->right,val);
        if(root->ra > root->right->ra) {
            rightRoate(root);
        }
    }
}

treapNode** find(treapNode *&root,int val) {
    if(!root) {
        return NULL;
    }
    if(val == root->val) {
        return &root;
    } else if(val < root->val) {
        return find(root->left,val);
    } else {
        return find(root->right,val);
    }
}

void erase(treapNode *&root) {
    if(!root->left) {
        treapNode* tmp = root;
        root = root->right;
        delete tmp;
        return;
    } 
    if(!root->right) {
        treapNode *tmp = root;
        root = root->left;
        delete tmp;
        return;
    }
    if(root->left->ra > root->right->ra) {
        rightRoate(root);
        erase(root->left);
    } else {
        leftRoate(root);
        erase(root->right);
    }
}


void destroyTreap(treapNode *&root,bool &first) {
    if(!root) {
        return;
    }
    destroyTreap(root->left,first);
    if(!first) printf(" ");
    printf("%d",root->val);
    first = false;
    destroyTreap(root->right,first);
    //root->left = root->right = NULL;
    //delete root;
}

int main() {
    int n;
    treapNode*root;
    while(scanf("%d",&n)==1) {
        root = NULL;
        for(int i=0,x;i<n;i++) {
            scanf("%d",&x);
            insert(root,x);
        }
        bool first = true;
        destroyTreap(root,first);
        printf("\n");
        for(int x;scanf("%d",&x) == 1;) {
            treapNode **tmp = find(root,x);
            if(!tmp) {
                printf("not exist\n");
                continue;
            }
            printf("find %d\n",(*tmp)->val);
            erase(*tmp);
            destroyTreap(root,first);
            printf("\n");
        }
    }
}

 

BST - treap

标签:erase   treap   des   span   lib   name   turn   ras   printf   

原文地址:http://www.cnblogs.com/cdyboke/p/7795301.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!