首页 > 代码库 > Treap模板
Treap模板
#include<bits/stdc++.h> #define inf 0x3f3f3f3f const int maxn=10000; using namespace std; struct node{ node* left; node* right; int value,fix,s,cnt; node(int _value,int _fix,int _s,int _cnt):value(_value),fix(_fix),s(_s),cnt(_cnt){} int lsize(){ if(left){ return left->s; } else { return 0; } } int rsize(){ if(right){ return right->s; } else{ return 0; } } void M(){ s=cnt; s+=lsize()+rsize(); } }; void left_rotate(node* &a){ node* b=a->right; a->right=b->left; b->left=a; a->M(); b->M(); a=b; } void right_rotate(node* &a){ node* b=a->left; a->left=b->right; b->right=a; a->M(); b->M(); a=b; } void treap_insert(node* &P,int value){ if(!P){ P=new node(value,rand(),1,1); } else{ if(P->value=http://www.mamicode.com/=value){ P->cnt++; } if(P->value>value){ treap_insert(P->left,value); } else { treap_insert(P->right,value); } if(P->fix>P->left->fix){ right_rotate(P); } else if(P->fix>P->right->fix){ left_rotate(P); } } P->M(); } void treap_delete(node* &P,int value){ if(P->value=http://www.mamicode.com/=value){ if(P->cnt==1){ if(!P->left||!P->right){ node* t=P; if(P->left) P=P->left; if(P->right) P=P->right; delete t; } else { if(P->left->fix<P->right->fix){ right_rotate(P); treap_delete(P->right,value); } else { left_rotate(P); treap_delete(P->left,value); } } } else{ P->cnt--; } } else { if(P->value>value){ treap_delete(P->left,value); } else { treap_delete(P->right,value); } } if(P){ P->M(); } } int treap_find(node* P,int value){ if(P->value=http://www.mamicode.com/=value){ return 1; } else { if(P->value>value){ if(P->left) treap_find(P->left,value); } else { if(P->right) treap_find(P->right,value); } } return 0; } int treap_kth(node* &P,int k){ int Lsize=P->lsize(); if(k<Lsize+1){ return treap_kth(P->left,k); } else if(k>Lsize+P->cnt){ return treap_kth(P->right,k-Lsize-P->cnt); } else { return P->value; } } int treap_rank(node* &P,int value,int cur){ int Lsize=P->lsize(); if(value=http://www.mamicode.com/=P->value){ return cur+Lsize+1; } else if(value<P->value){ return treap_rank(P->left,value,cur); } else { return treap_rank(P->right,value,cur+Lsize+P->cnt); } } node* tpred(node* &P,int value,node* &best){ if(!P){ return best; } else { if(P->value<value){ return tpred(P->right,value,P); } else{ return tpred(P->left,value,best); } } } node* tsuc(node* &P,int value,node* &best){ if(!P){ return best; } else { if(P->value>value){ return tsuc(P->left,value,P); } else { return tsuc(P->right,value,best); } } } int main() { srand(time(0)); return 0; }
Treap模板
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。