AVL添加(c语言)

平衡二叉树,是一种二叉排序树,其中每个结点的左子树和右子树的高度差至多等于1。它是一种高度平衡的二叉排序树。现二叉平衡树结点定义如下:

typedef struct node
{
    int val;
    struct node *left;
    struct node *right;
    struct node *parent;
    int height;
} node_t;

请实现平衡二叉树的插入算法:

//向根为 root 的平衡二叉树插入新元素 val,成功后返回新平衡二叉树根结点
node_t *avl_insert(node_t *root, int val);

答案:
注意:对于这个parent弄得不是很清楚,希望有大佬可以指点一下。其实没有parent也能过。。。

node_t* Insert(node_t* T, int X);
int GetHeight(node_t* T);
int Max(int a, int b);
node_t* SingleLeftRotation(node_t* A);
node_t* DoubleLeftRightRotation(node_t* A);
node_t* SingleRightRotation(node_t* A);
node_t* DoubleRightLeftRotation(node_t* A);

int Max(int a, int b)
{
    return a > b ? a : b;
}

int GetHeight(node_t* T)
{
    if (T)
        return T->height;
    else
        return 0;
}

node_t* SingleLeftRotation(node_t* A) //左单旋
{
    node_t* B;
    B = A->left;
    A->left = B->right;
    B->right = A;
    A->height = Max(GetHeight(A->left), GetHeight(A->right)) + 1;
    B->height = Max(GetHeight(B->left), GetHeight(B->right)) + 1;
    return B;
}

node_t* DoubleLeftRightRotation(node_t* A) //左右双旋
{
    A->left = SingleRightRotation(A->left);
    return SingleLeftRotation(A);
}

node_t* SingleRightRotation(node_t* A) //右单旋
{
    node_t* B;
    B = A->right;
    A->right = B->left;
    B->left = A;
    A->height = Max(GetHeight(A->left), GetHeight(A->right)) + 1;
    B->height = Max(GetHeight(B->left), GetHeight(B->right)) + 1;
    return B;
}

node_t* DoubleRightLeftRotation(node_t* A) //右左双旋
{
    A->right = SingleLeftRotation(A->right);
    return SingleRightRotation(A);
}

node_t* avl_insert(node_t* root, int val)
{
    node_t* parentNode = NULL;
    if (!root) {
        root = (node_t*)malloc(sizeof(node_t));
        root->left = root->right = root->parent = NULL;
        root->val = val;
        root->height = 1;
    } else if (val < root->val) {
        parentNode = root->left;
        root->left = avl_insert(root->left, val);
        if (GetHeight(root->left) - GetHeight(root->right) == 2) {
            if (val < root->left->val)
                root = SingleLeftRotation(root);
            else {
                root = DoubleLeftRightRotation(root);
            }
        }
    } else if (val > root->val) {
        parentNode = root->left;
        root->right = avl_insert(root->right, val);
        if (GetHeight(root->right) - GetHeight(root->left) == 2) {
            if (val > root->right->val)
                root = SingleRightRotation(root);
            else
                root = DoubleRightLeftRotation(root);
        }
    }
    root->parent = parentNode;
    root->height = Max(GetHeight(root->left), GetHeight(root->right)) + 1;
    return root;
}
上一篇:AVL平衡二叉树C++代码实现


下一篇:iis配置错误