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
方法打印出树的节点值,以便观察旋转操作的效果。