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
方法打印出樹的節點值,以便觀察旋轉操作的效果。