温馨提示×

java treenode能实现节点旋转吗

小樊
84
2024-11-28 09:48:41
栏目: 编程语言

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);
    }
}

在这个示例中,我们定义了两个方法leftRotaterightRotate,分别用于实现左旋和右旋操作。在main方法中,我们创建了一个简单的二叉树,并对其进行了左旋和右旋操作。最后,我们使用printTree方法打印出树的节点值,以便观察旋转操作的效果。

0