java Treenode工具类

原创

mob64ca12d2dee8 2024-08-30 06:22:31 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12d2dee8的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java Treenode工具类科普

在数据结构中，树是一种常见的非线性结构。二叉树是树的一种重要类型，每个节点最多有两个子节点。在Java中，正确地处理树结构可以帮助我们有效地解决许多算法和数据管理问题。本文将从基础概念出发，深入探讨Java中的TreeNode工具类，并提供相应的代码示例。

一、TreeNode的定义

TreeNode通常用于表示二叉树中的节点。一个基本的TreeNode类通常包含三个成员变量：节点值、左子节点和右子节点。以下是TreeNode的基本实现：

public class TreeNode {
    int val;         // 节点的值
    TreeNode left;  // 左子节点
    TreeNode right; // 右子节点

    // 构造函数
    public TreeNode(int val) {
        this.val = val;
        this.left = null;
        this.right = null;
    }
}

二、树的基本操作

常见的树操作包括：插入节点、遍历树（前序遍历、中序遍历、后序遍历）、查找节点和删除节点等。下面将逐一展现这些操作的代码实现。

1. 插入节点

在二叉搜索树中，节点的左子树中的所有节点值都小于该节点的值，而右子树中的节点值都大于该节点的值。下面是节点插入的实现代码：

public TreeNode insert(TreeNode root, int val) {
    if (root == null) {
        return new TreeNode(val);
    }
    if (val < root.val) {
        root.left = insert(root.left, val);
    } else {
        root.right = insert(root.right, val);
    }
    return root;
}

2. 遍历树

树的遍历是理解树结构的关键。我们可以通过递归或迭代的方式实现前序、中序和后序遍历。以下是前序遍历的示例：

public void preOrderTraversal(TreeNode root) {
    if (root != null) {
        System.out.print(root.val + " "); // 访问根节点
        preOrderTraversal(root.left);      // 遍历左子树
        preOrderTraversal(root.right);     // 遍历右子树
    }
}

3. 查找节点

在二叉搜索树中查找节点十分简单，只需根据节点值进行比较，沿着树的方向进行查找：

public TreeNode search(TreeNode root, int val) {
    if (root == null || root.val == val) {
        return root;
    }
    if (val < root.val) {
        return search(root.left, val);
    } else {
        return search(root.right, val);
    }
}

4. 删除节点

删除节点稍微复杂一些。我们要考虑三种情况：删除的节点没有子节点、有一个子节点、和有两个子节点。

public TreeNode deleteNode(TreeNode root, int val) {
    if (root == null) return null;
    
    if (val < root.val) {
        root.left = deleteNode(root.left, val);
    } else if (val > root.val) {
        root.right = deleteNode(root.right, val);
    } else {
        // 节点有两个子节点
        if (root.left != null && root.right != null) {
            TreeNode minNode = findMin(root.right);
            root.val = minNode.val;
            root.right = deleteNode(root.right, minNode.val);
        } else {
            // 节点只有一个子节点或没有子节点
            root = (root.left != null) ? root.left : root.right;
        }
    }
    return root;
}

private TreeNode findMin(TreeNode node) {
    while (node.left != null) {
        node = node.left;
    }
    return node;
}

三、状态图与类图

在实现树结构和相应操作时，了解类与对象的关系以及状态的转变是十分重要的。结合Mermaid语法，我们可以呈现状态图与类图。

状态图

stateDiagram
    [*] --> Empty
    Empty --> Node
    Node --> Inserted
    Node --> Deleted
    Inserted --> Search
    Deleted --> [*]

类图

classDiagram
    class TreeNode {
        int val
        TreeNode left
        TreeNode right
        +TreeNode(int val)
        +insert(TreeNode root, int val)
        +preOrderTraversal(TreeNode root)
        +search(TreeNode root, int val)
        +deleteNode(TreeNode root, int val)
    }