程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> HDU 4585 平衡樹Treap

HDU 4585 平衡樹Treap

編輯:關於C++

題意:給出n組數,第一個數是id,第二個數是級別,每輸入一個,輸出這個人和哪個人打架,這個人會找和他級別最相近的人打,如果有兩個人級別和他相差的一樣多,他就會選擇級別比他小的打架。

思路:用treap完成,可以用STL水過,但要練Treap就寫了平衡樹的,對於每個人的等級,我們找到他的等級的排名t,然後找第t-1大的數和第t+1大的數,然後進行比較一下。討論後輸出,PS:沒有好得模版,只能在網上學了一個還可以接受的.......

#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=100010;
class Treap{
    public:
struct Treap_Node{
    Treap_Node *left,*right;
    int value,fix,weight,size;//fix為修正值,是隨機產生的,保證二叉樹不會退化的重要環節,wweight為權值,size為子樹大小
}*root,*null;
Treap(){
    null=new struct Treap_Node;
    null->size=0;
    null->weight=0;
    null->value=inf;
    null->left=null;
    null->right=null;
    null->fix=inf;
    root=null;
}
void Treap_Print(Treap_Node *P){//從小到大輸出
    if(P!=null){
        Treap_Print(P->left);
        printf("%d\n",P->value);
        Treap_Print(P->right);
    }
}
void Treap_Left_Rotate(Treap_Node *&a){//左旋之後仍不改變二叉樹性質
    Treap_Node *b=a->right;
    a->right=b->left;
    b->left=a;
    b->size=a->size;
    a->size=a->left->size+a->right->size+a->weight;
    a=b;
}
void Treap_Right_Rotate(Treap_Node *&a){//右旋之後仍不改變二叉樹性質
    Treap_Node *b=a->left;
    a->left=b->right;
    b->right=a;
    b->size=a->size;
    a->size=a->left->size+a->right->size+a->weight;
    a=b;
}
int Treap_Find(Treap_Node *P,int value){//查找有沒有value這個數
    if(P==null) return 0;
    if(P->value==value) return 1;
    else if(valuevalue) return Treap_Find(P->left,value);
    else return Treap_Find(P->right,value);
}
void Treap_Insert(Treap_Node *&P,int value){//插入一個數
    if(P==null){
        P=new Treap_Node;
        P->left=P->right=null;//左右兒子均為空
        P->value=value;
        P->fix=rand();
        P->weight=1;
        P->size=1;
    }else if(value==P->value){
        P->weight++;
    }
    else if(valuevalue){
        Treap_Insert(P->left,value);
        if(P->left->fixfix)
            Treap_Right_Rotate(P);
    }else{
        Treap_Insert(P->right,value);
        if(P->right->fixfix)
            Treap_Left_Rotate(P);
    }
    P->size=P->left->size+P->right->size+P->weight;
}
void Treap_Delete(Treap_Node *&P,int value){//刪除一個數
    if(P==null) return ;
    if(valuevalue) Treap_Delete(P->left,value);
    else if(value>P->value) Treap_Delete(P->right,value);
    else if(P->weight>1) P->weight--;
    else if((P->left==NULL)&&(P->right==NULL)){
        delete P;
        P=NULL;
    }else{
        if(P->left->fixright->fix) Treap_Left_Rotate(P);
        else Treap_Right_Rotate(P);
        Treap_Delete(P,value);
    }
    P->size=P->left->size+P->right->size+P->weight;
}
int Treap_pred(Treap_Node *P,int value,Treap_Node *optimal){
    if(P==null||value==P->value) return optimal->value;
    if(P->valueright,value,P);
    else return Treap_pred(P->left,value,optimal);
}
int Treap_succ(Treap_Node *P,int value,Treap_Node *optimal){
    if(P==null||value==P->value) return optimal->value;
    if(P->value>value) return Treap_succ(P->left,value,P);
    else return Treap_succ(P->right,value,optimal);
}
int Treap_Findkth(Treap_Node *P,int k){//求第K大的數
    if(P==null) return 0;
    int t=P->left->size;
    if(kleft,k);
    else if(k>t+P->weight) return Treap_Findkth(P->right,k-(t+P->weight));
    else return P->value;
}
int Treap_Rank(Treap_Node *P,int value,int cur){
    int t=P->left->size;
    if(value==P->value) return t+cur+1;
    else if(valuevalue) return Treap_Rank(P->left,value,cur);
    else return Treap_Rank(P->right,value,t+cur+P->weight);
}
void Treap_erase(Treap_Node *&P) {
		if(P->left!=null)
			Treap_erase(P->left);
		if(P->right!=null)
			Treap_erase(P->right);
		delete P;
}
};
int id[5000010];
int main(){
    int n,a,b;
    while(scanf("%d",&n)!=-1){
        if(n==0) break;
        Treap tree;
        int ans1,ans2;
        memset(id,0,sizeof(id));
        scanf("%d%d",&a,&b);
        printf("%d 1\n",a);
        id[b]=a;tree.Treap_Insert(tree.root,b);
        for(int i=1;i

 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved