1. 程式人生 > >平衡二叉樹的建立 c語言範例

平衡二叉樹的建立 c語言範例

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct node
{
    int data;
    int high;
    struct node *lt, *rt;
};

int max(int a, int b)
{
    return a>b?a:b;
}
///深度
int deep(struct node *root)
{
    if(root==NULL)
        return -1;
    else return root->high;
}

struct node *LL(struct node *root)   // 左兒子變為根, 根變為左兒子的右兒子. 
{
    struct node *p=root->lt;
    root->lt=p->rt;
    p->rt=root;
    root->high=max(deep(root->lt), deep(root->rt))+1;
    return p;
}

struct node *RR(struct node *root)  // 右兒子變為根, 根變為右兒子的左兒子. 
    struct node *p=root->rt;
    root->rt=p->lt;
    p->lt=root;
    root->high=max(deep(root->lt), deep(root->rt))+1;
    return p;
}

struct node *RL(struct node *root)
{
    root->rt=LL(root->rt);
    root=RR(root);
    return root;
}

struct node *LR(struct node *root)
{
    root->lt=RR(root->lt);
    root=LL(root);
    return root;
}

struct node *Insert(struct node *root, int num)
{
    if(!root)
    {
        root=(struct node *)malloc(sizeof(struct node));
        root->data=num;
        root->high=0;
        root->lt=NULL;
        root->rt=NULL;
    }
    else
    {
        if(num<root->data)///資料比根小
        {
            root->lt=Insert(root->lt, num);///加到左子樹上

            ///先判斷是否要轉
            if(abs(deep(root->lt)-deep(root->rt))>1)//深度之差大於一就轉
            {
               if(num>=root->lt->data)///比左子根大,轉兩次,即LR
                {
                    root=LR(root);
                }
                else ///比左子根小,轉一次,即LL
                {
                    root=LL(root);
                }
            }

        }
        else///比根大
        {
            root->rt=Insert(root->rt, num);///加到右子樹上
            if(abs(deep(root->lt)-deep(root->rt))>1)
            {
                if(num>=root->rt->data)///加到右子樹的右邊,轉一次,RR
                {
                    root=RR(root);
                }
                else///加到右子樹的左邊,轉兩次,RL
                {
                    root=RL(root);
                }
            }
        }
    }

    root->high=max(deep(root->lt), deep(root->rt))+1;
    printf("node-data=%d, node-high=%d\n",root->data,root->high);
    return root;
}
int main()
{
    struct node *root=NULL;

    root=Insert(root,23);
    printf("root-data=%d\n", root->data);
    root=Insert(root,10);
    printf("root-data=%d\n", root->data);
    root=Insert(root,17);
    printf("root-data=%d\n", root->data);
    root=Insert(root,73);
    printf("root-data=%d\n", root->data);
    root=Insert(root,8);
    printf("root-data=%d\n", root->data);
    root=Insert(root,11);
    root=Insert(root,46);
    root=Insert(root,27);
    root=Insert(root,6);
    root=Insert(root,3);
    root=Insert(root,10);
    printf("root-data=%d\n", root->data);