Java實現樹形菜單的方法總結

更新時間：2023年08月30日 10:46:40 作者：一只愛擼貓的程序猿

當我們想要展示層級結構,如文件目錄、組織結構或分類目錄時,樹形菜單是一個直觀且有效的解決方案,本文為大家整理了java中幾種常見方法,希望對大家有所幫助

1. 使用遞歸法

遞歸法是最直觀的實現樹形菜單的方法。我們先定義節(jié)點結構，然后通過遞歸方式構建每一個節(jié)點的子節(jié)點。

實例代碼：

class TreeNode {
    int id;
    int parentId;
    String name;
    List<TreeNode> children;
    // 構造函數、getters、setters省略
}
public List<TreeNode> buildTree(List<TreeNode> nodes, int parentId) {
    List<TreeNode> tree = new ArrayList<>();
    for (TreeNode node : nodes) {
        if (node.parentId == parentId) {
            node.children = buildTree(nodes, node.id);
            tree.add(node);
        }
    }
    return tree;
}

2. 使用隊列法 (層次遍歷)

使用隊列實現樹形菜單可以有效減少遞歸的深度，特別是在樹形結構很深的情況下。

實例代碼：

public List<TreeNode> buildTreeWithQueue(List<TreeNode> nodes) {
    if (nodes == null || nodes.isEmpty()) {
        return Collections.emptyList();
    }
    Map<Integer, TreeNode> nodeMap = nodes.stream().collect(Collectors.toMap(TreeNode::getId, node -> node));
    List<TreeNode> tree = new ArrayList<>();
    Queue<TreeNode> queue = new LinkedList<>(nodes);
    while (!queue.isEmpty()) {
        TreeNode node = queue.poll();
        if (node.parentId == 0) {
            tree.add(node);
        } else {
            TreeNode parent = nodeMap.get(node.parentId);
            if (parent.children == null) {
                parent.children = new ArrayList<>();
            }
            parent.children.add(node);
        }
    }
    return tree;
}

3. 使用Map索引優(yōu)化

通過使用Map進行索引，可以提高搜索效率，特別是當節(jié)點數量非常多時。

實例代碼：

public List<TreeNode> buildTreeWithMap(List<TreeNode> nodes) {
    Map<Integer, List<TreeNode>> childrenMap = new HashMap<>();
    List<TreeNode> rootNodes = new ArrayList<>();
    for (TreeNode node : nodes) {
        if (node.parentId == 0) {
            rootNodes.add(node);
        } else {
            childrenMap
                .computeIfAbsent(node.parentId, k -> new ArrayList<>())
                .add(node);
        }
    }
    for (TreeNode rootNode : rootNodes) {
        rootNode.children = getChildren(rootNode.id, childrenMap);
    }
    return rootNodes;
}
private List<TreeNode> getChildren(int parentId, Map<Integer, List<TreeNode>> childrenMap) {
    List<TreeNode> children = childrenMap.get(parentId);
    if (children != null) {
        for (TreeNode child : children) {
            child.children = getChildren(child.id, childrenMap);
        }
    }
    return children;
}

4. 數據庫的樹形編碼的方式

步驟1: 創(chuàng)建數據庫表結構

首先，我們需要一個數據庫表來存儲節(jié)點數據：

CREATE TABLE tree_nodes (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(255) NOT NULL,
    path VARCHAR(255) DEFAULT '',
    parent_id INT
);

其中，path字段用于存儲從根節(jié)點到當前節(jié)點的路徑，如1/3/5/。

步驟2: Java 數據模型

我們需要一個Java類來表示數據庫中的節(jié)點：

public class TreeNode {
    private int id;
    private String name;
    private String path;
    private int parentId;
    // Getter, Setter 和 Constructor
}

步驟3: 插入節(jié)點

插入一個新的節(jié)點時，我們需要計算它的路徑：

public void addNode(String name, int parentId, Connection connection) throws SQLException {
    String path;
    if (parentId == 0) {
        path = "/";
    } else {
        String parentPathQuery = "SELECT path FROM tree_nodes WHERE id = ?";
        try (PreparedStatement stmt = connection.prepareStatement(parentPathQuery)) {
            stmt.setInt(1, parentId);
            ResultSet rs = stmt.executeQuery();
            if (rs.next()) {
                path = rs.getString("path") + parentId + "/";
            } else {
                throw new SQLException("Parent not found");
            }
        }
    }
    String insertQuery = "INSERT INTO tree_nodes (name, path, parent_id) VALUES (?, ?, ?)";
    try (PreparedStatement stmt = connection.prepareStatement(insertQuery)) {
        stmt.setString(1, name);
        stmt.setString(2, path);
        stmt.setInt(3, parentId);
        stmt.executeUpdate();
    }
}

步驟4: 獲取某節(jié)點的所有子節(jié)點

public List<TreeNode> getChildren(int nodeId, Connection connection) throws SQLException {
    List<TreeNode> children = new ArrayList<>();
    String pathQuery = "SELECT path FROM tree_nodes WHERE id = ?";
    String path;
    try (PreparedStatement stmt = connection.prepareStatement(pathQuery)) {
        stmt.setInt(1, nodeId);
        ResultSet rs = stmt.executeQuery();
        if (rs.next()) {
            path = rs.getString("path") + nodeId + "/";
        } else {
            throw new SQLException("Node not found");
        }
    }
    String childrenQuery = "SELECT * FROM tree_nodes WHERE path LIKE ?";
    try (PreparedStatement stmt = connection.prepareStatement(childrenQuery)) {
        stmt.setString(1, path + "%");
        ResultSet rs = stmt.executeQuery();
        while (rs.next()) {
            TreeNode node = new TreeNode();
            node.setId(rs.getInt("id"));
            node.setName(rs.getString("name"));
            node.setPath(rs.getString("path"));
            node.setParentId(rs.getInt("parent_id"));
            children.add(node);
        }
    }
    return children;
}

總結

以上我們介紹了三種在Java中實現樹形菜單的方法，從基礎的遞歸法、隊列法到使用Map進行優(yōu)化。根據實際需求和數據量大小，你可以選擇合適的方法進行實現。不過，無論使用哪種方法，都要確保理解樹形結構的基礎原理和具體實現細節(jié)，以確保代碼的正確性和效率。還有一種是路徑枚舉方式的一個簡單例子，它已經足夠用于實際應用。當然，還有許多細節(jié)可以優(yōu)化，如路徑長度限制、節(jié)點移動等。而其他的樹形編號方法，如嵌套集模型和閉包表，會有更復雜的邏輯。但總的來說，理解核心概念并將其應用于實際代碼是關鍵。

到此這篇關于Java實現樹形菜單的方法總結的文章就介紹到這了,更多相關Java樹形菜單內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: