BST - treap

Posted 2020-10-13 dy blog

#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");
        }
    }
}