Java中的TreeNode类通常用于表示二叉树的节点。要实现节点旋转,你需要编写一些额外的代码。以下是一个简单的示例,展示了如何在Java中实现二叉树的左旋和右旋操作。
首先,定义一个TreeNode类:
public class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } }
接下来,实现左旋和右旋操作:
public class BinaryTreeRotation { // 左旋操作 public static TreeNode leftRotate(TreeNode x) { TreeNode y = x.right; x.right = y.left; y.left = x; return y; } // 右旋操作 public static TreeNode rightRotate(TreeNode y) { TreeNode x = y.left; y.left = x.right; x.right = y; return x; } public static void main(String[] args) { TreeNode root = new TreeNode(1); root.left = new TreeNode(2); root.right = new TreeNode(3); root.left.left = new TreeNode(4); root.left.right = new TreeNode(5); System.out.println("Original tree:"); printTree(root); root = leftRotate(root); System.out.println("\nTree after left rotation:"); printTree(root); root = rightRotate(root.left); System.out.println("\nTree after right rotation:"); printTree(root); } // 打印树结构 public static void printTree(TreeNode root) { if (root == null) { return; } System.out.print(root.val + " "); printTree(root.left); printTree(root.right); } }
在这个示例中,我们定义了两个方法leftRotate
和rightRotate
,分别用于实现左旋和右旋操作。在main
方法中,我们创建了一个简单的二叉树,并对其进行了左旋和右旋操作。最后,我们使用printTree
方法打印出树的节点值,以便观察旋转操作的效果。