Java中如何构建平衡二叉树-编程知识

Java中如何构建平衡二叉树

定义：平衡二叉树是一棵二叉排序树，或者为空，或者满足以下条件：

1)左右子树高度差的绝对值不大于1；

2)左右子树都是平衡二叉树。

平衡因子：左子树的高度减去右子树的高度，显然，在平衡二叉树中，每个结点的平衡因子的值为-1，0或1。

在Java中，构造平衡二叉树通常涉及到 AVL 树（Adelson-Velsky and Landis tree）的实现。AVL 树是一种自平衡二叉搜索树，它通过保持每个节点的左右子树高度差不超过1来保持平衡。

插入操作的可能情况：

LL型：新插入结点在A的左孩子(L)的左子树(L)中；

LR型：新插入结点在A的左孩子(L)的右子树(R)中；

RL型：新插入结点在A的右孩子(R)的左子树(L)中；

RR型：新插入结点在A的右孩子(R)的右子树(R)中。

LL型：右旋是指将一个节点的左子树变为其父节点，同时该节点成为其左子树的右子节点。右旋用于解决左子树过深的情况。

LR型：右左旋是先对当前节点的右子树进行右旋，然后再对当前节点进行左旋。这用于解决右子树的左子树过深的情况。

RL型：左右旋是先对当前节点的左子树进行左旋，然后再对当前节点进行右旋。这用于解决左子树的右子树过深的情况。

RR型：左旋是指将一个节点的右子树变为其父节点，同时该节点成为其右子树的左子节点。左旋用于解决右子树过深的情况。

以上都是基于左旋和右旋实现的，那么接下来我们构建左旋和右旋的方法以及插入数据对树结构所做操作：

对一组数据构建平衡二叉树：10、5、20、13、17、4

现在我们利用代码实现。

首先，需要创建一个 TreeNode 类，这个类除了包含节点值之外，还需要保存节点的高度：

public class TreeNode {int val;int height;TreeNode left;TreeNode right;public TreeNode(int val) {this.val = val;this.height = 1; // 初始高度为1}
}

然后，创建一个 AVL 树的实现类，其中包括插入操作和平衡操作：

public class AVLTree {private TreeNode root;// 获取节点的高度private int height(TreeNode node) {if (node == null) {return 0;}return node.height;}// 获取平衡因子private int getBalance(TreeNode node) {if (node == null) {return 0;}return height(node.left) - height(node.right);}// 更新节点的高度private void updateHeight(TreeNode node) {if (node != null) {node.height = Math.max(height(node.left), height(node.right)) + 1;}}// 右旋转private TreeNode rightRotate(TreeNode y) {TreeNode x = y.left;TreeNode T2 = x.right;// 执行旋转x.right = y;y.left = T2;// 更新高度updateHeight(y);updateHeight(x);return x;}// 左旋转private TreeNode leftRotate(TreeNode x) {TreeNode y = x.right;TreeNode T2 = y.left;// 执行旋转y.left = x;x.right = T2;// 更新高度updateHeight(x);updateHeight(y);return y;}// 插入节点public TreeNode insert(TreeNode node, int val) {if (node == null) {return new TreeNode(val);}// 执行标准的BST插入if (val < node.val) {node.left = insert(node.left, val);} else if (val > node.val) {node.right = insert(node.right, val);} else {// 重复的值不被允许return node;}// 更新节点的高度updateHeight(node);// 获取平衡因子int balance = getBalance(node);// 左子树的左侧插入，需要右旋转if (balance > 1 && val < node.left.val) {return rightRotate(node);}// 右子树的右侧插入，需要左旋转if (balance < -1 && val > node.right.val) {return leftRotate(node);}// 左子树的右侧插入，需要先左旋转再右旋转if (balance > 1 && val > node.left.val) {node.left = leftRotate(node.left);return rightRotate(node);}// 右子树的左侧插入，需要先右旋转再左旋转if (balance < -1 && val < node.right.val) {node.right = rightRotate(node.right);return leftRotate(node);}return node;}// 插入新节点，外部调用该方法public TreeNode insert(int val) {root = insert(root, val);//返回根节点return root;}}

定义一个BinaryTreePrinter类，使用深度优先搜索来打印二叉树：

public class BinaryTreePrinter {public static void printTree(TreeNode root) {printTreeHelper(root, 0);}private static void printTreeHelper(TreeNode node, int depth) {if (node == null) {return;}// 打印右子树，使其显示在上方printTreeHelper(node.right, depth + 1);// 打印当前节点for (int i = 0; i < depth; i++) {System.out.print("    "); // 每层缩进四个空格}System.out.println(node.val);// 打印左子树，使其显示在下方printTreeHelper(node.left, depth + 1);}
}

在测试类Test中创建了一个 AVLTree 实例，并调用其方法来操作 AVL 树：

public class Test {public static void main(String[] args) {AVLTree avlTree =new AVLTree();// 构造一棵平衡二叉树      avlTree.insert(5);avlTree.insert(10);avlTree.insert(20);avlTree.insert(13);avlTree.insert(17);TreeNode root=avlTree.insert(12);System.out.println("平衡二叉树高度:"+root.height);BinaryTreePrinter.printTree(root);}
}

根节点为13，结果如下：