树形菜单 java 查询

树形菜单查询

树形菜单是一种常见的显示数据层级关系的方式，通常用于展示组织结构、文件目录、分类等数据。在Java中，我们可以使用递归算法来实现树形菜单的查询和展示。本文将介绍如何使用Java实现树形菜单的查询，并附带代码示例。

流程图

下面是树形菜单查询的基本流程图：

flowchart TD
    A(开始)
    B(查询根节点)
    C(递归查询子节点)
    D(添加子节点)
    E(返回结果)
    F(结束)
    A-->B
    B-->C
    C-->D
    D-->C
    C-->E
    E-->F

示例代码

首先，我们需要定义一个树节点类TreeNode，它包含节点ID和父节点ID两个属性，以及一个子节点列表。

class TreeNode {
    private int id;
    private int parentId;
    private List<TreeNode> children;

    // 省略构造函数和getter/setter方法

    public void addChild(TreeNode child) {
        if (children == null) {
            children = new ArrayList<>();
        }
        children.add(child);
    }
}

接下来，我们编写树形菜单查询的方法：

class TreeMenu {
    public List<TreeNode> queryMenu(List<TreeNode> nodes, int parentId) {
        List<TreeNode> result = new ArrayList<>();
        for (TreeNode node : nodes) {
            if (node.getParentId() == parentId) {
                result.add(node);
                List<TreeNode> children = queryMenu(nodes, node.getId());
                node.setChildren(children);
            }
        }
        return result;
    }
}

上述代码使用递归算法查询菜单，首先查询根节点，然后递归查询该节点的子节点，并将子节点添加到父节点的子节点列表中。

最后，我们可以使用下面的代码示例来测试树形菜单的查询：

public class Main {
    public static void main(String[] args) {
        // 创建一些示例节点
        TreeNode node1 = new TreeNode(1, 0);
        TreeNode node2 = new TreeNode(2, 0);
        TreeNode node3 = new TreeNode(3, 1);
        TreeNode node4 = new TreeNode(4, 1);
        TreeNode node5 = new TreeNode(5, 2);

        List<TreeNode> nodes = new ArrayList<>();
        nodes.add(node1);
        nodes.add(node2);
        nodes.add(node3);
        nodes.add(node4);
        nodes.add(node5);

        TreeMenu treeMenu = new TreeMenu();
        List<TreeNode> menu = treeMenu.queryMenu(nodes, 0);

        // 打印查询结果
        printMenu(menu);
    }

    private static void printMenu(List<TreeNode> menu) {
        for (TreeNode node : menu) {
            System.out.println("节点ID：" + node.getId());
            if (node.getChildren() != null) {
                printMenu(node.getChildren());
            }
        }
    }
}

运行上述代码，我们将得到如下输出：

节点ID：1
节点ID：3
节点ID：4
节点ID：2
节点ID：5

总结

本文介绍了如何使用Java实现树形菜单的查询，通过递归算法可以轻松地查询并展示树形数据。我们可以根据实际需求，进一步扩展树节点类和查询方法，以满足不同的场景。

希望本文能帮助读者理解树形菜单的查询原理，并能在实际项目中应用。如有疑问，欢迎留言讨论。