将BST与RR LL LR RL的四种情况进行平衡

时间:2014-07-10 13:26:26

标签: c binary-search-tree

我写了一个平衡BST的代码,但是我不能在BST中添加超过5个元素! 我们有3种情况,我们从不平衡的节点到我们要插入它的节点的RR,当我添加5个节点时这个代码工作正常..但是如果我添加更多它会给我overStack .. 这是代码

#include"iostream"
using namespace std;
struct node{
    int x;
    node*left;
    node*right;
};
int max(int x,int y){
    if(x>y)return x;return y;
}
int depth(node*T){ // returns the depth of the BST
    if(!T)return 0;
    return max(depth(T->left),depth(T->right))+1;
}
int isBalance(node*T){   //check if the BST is Balanced or not
    return abs(depth(T->left)-depth(T->right))<=1;
}
int Expresion(char*x){ // for switch case
    if(!strcmp(x,"RR"))return 3;
    if(!strcmp(x,"LL"))return 2;
    if(!strcmp(x,"RL"))return 1;
    if(!strcmp(x,"LR"))return 0;
    return -1;
}
node* getBalance(node*T,char*x){   //Balance the node
    node*temp=NULL;
    node*temp1=NULL;
    node*z=NULL;
    switch(Expresion(x)){
    case 0:
        temp=T->left;
        temp1=temp->right;
        z = (temp1->right!=NULL)?temp1->right:temp1->left;
        temp1->right=T;
        T->left=z;
        temp->right=NULL;
        return temp1;
        break;
    case 1:
        temp=T->right;
        temp1=temp->left;
        z = (temp1->right!=NULL)?temp1->right:temp1->left;
        temp1->left=T;
        T->right=z;
        temp->left=NULL;
        return temp1;
        break;
    case 2:
        temp=T->left;
        temp1=temp->left;
        temp->right=T;
        temp->left=temp1;
        return temp;
        break;
    case 3:
        temp=T->right;
        temp1=temp->right;
        temp->left=T;
        temp->right=temp1;
        return temp;
        break;
    }
    return NULL;
}
char *Equation(node*x){ //find what is the situation (RR , LL , RL , LR)
    char*z=new char;z[0]=NULL;
    if(depth(x->right) > depth(x->left)){strcat(z,"R"); x=x->right;}
    else {strcat(z,"L");x=x->left;}
    if(depth(x->right) > depth(x->left)){strcat(z,"R"); x=x->right;}
    else {strcat(z,"L");x=x->left;}
    return z;
}
void balance (node*&T){ //balance the BST
    if(!T)return;
    if(!isBalance(T)){
        T=getBalance(T,Equation(T));
        return;
    }
    balance(T->left);
    balance(T->right);
}
node*insert1(node*T,int x){ //insert in the BST
    if(!T){node*temp=new node;
    temp->x=x;temp->left=temp->right=NULL;return temp;}
    if(T->x > x) T->left=insert1(T->left,x);
    else{T->right=insert1(T->right,x);}
    return T;
}
node* insert(node*T,int x){ //insert then Balance the BST
    T=insert1(T,x);
    balance(T);
    return T;
}
void print(node*T){ //prints the BST
    if(!T)return;
    cout<<T->x<<"  ";
    print(T->right);
    print(T->left);
}
void main(){
    node*head=NULL;
    /*for(int i=0;i<3;i++){
        if(i>50) head=insert(head,(i-3)/2);
        else head=insert(head,(i+3)*2);
    }*/
    head=insert(head,2);
    head=insert(head,4);
    head=insert(head,1);
    head=insert(head,0);
    head=insert(head,10);
    head=insert(head,64);
    head=insert(head,123);
    head=insert(head,121);
    head=insert(head,28);
    head=insert(head,12);
    head=insert(head,240);
    head=insert(head,42);
    head=insert(head,142);
    head=insert(head,76);
    head=insert(head,100);
    cout<<isBalance(head)<<endl;
    //print(head);
    system("pause");
}

1 个答案:

答案 0 :(得分:0)

你的树在某处有一个循环引用:

  • (节点4) - &gt;右指向(节点10)
  • (节点10) - &gt;左点指向(节点4)

然后你尝试平衡树和你的程序扼流圈计算它的深度:无限递归。

这是插入节点123后的树(平衡时):

root: (Node 2)

(Node 2)->left: (Node 1)
 (Node 1)->left: (Node 0)
 (Node 1)->right: none

(Node 2)->right: (Node 10)
 (Node 10)->left: (Node 4)
  (Node 4)->left: none
  (Node 4)->right: (Node 10) !!!!

 (Node 10)->right: (Node 64)
  (Node 64)->left: none
  (Node 64)->right: (Node 123)

我建议您创建一个显示整个树的函数,并在每个insert之后调用它。您的插入路径(insert本身或balance)有一个错误。