Java中遞歸構(gòu)建樹形結(jié)構(gòu)的算法解讀

更新時間：2025年03月04日 08:48:50 作者：微笑聽雨。

該文章介紹了如何使用Java遞歸算法構(gòu)建樹形結(jié)構(gòu),通過定義樹節(jié)點類,遍歷扁平數(shù)據(jù)列表,將節(jié)點加入對應父節(jié)點的子節(jié)點列表中,實現(xiàn)從扁平數(shù)據(jù)到樹形結(jié)構(gòu)的轉(zhuǎn)換

Java遞歸構(gòu)建樹形結(jié)構(gòu)的算法

java遞歸獲取樹

當我們處理像部門樹、菜單樹等樹形結(jié)構(gòu)數(shù)據(jù)時，我們需要從后端獲取一組扁平的數(shù)據(jù)列表。

這些數(shù)據(jù)列表通常由一個父節(jié)點ID字段和一個當前節(jié)點ID字段組成，這樣我們可以通過這兩個字段建立節(jié)點之間的父子關(guān)系。然后，我們可以利用這些數(shù)據(jù)，通過遞歸的方式構(gòu)建樹形結(jié)構(gòu)。

在Java后端中，我們可以通過遞歸函數(shù)來實現(xiàn)這一過程。
首先，我們需要定義一個樹節(jié)點類，該類包含節(jié)點的基本信息和一個子節(jié)點列表。
然后，我們遍歷扁平數(shù)據(jù)列表，對于每一個節(jié)點，我們將其加入到對應父節(jié)點的子節(jié)點列表中。
如果當前節(jié)點沒有子節(jié)點，我們可以直接將其加入到樹的根節(jié)點列表中。
然后，我們遞歸處理當前節(jié)點的子節(jié)點，直到所有節(jié)點都被加入到樹中。

通過這種方式，我們可以將扁平的數(shù)據(jù)列表轉(zhuǎn)換成樹形結(jié)構(gòu)，使得數(shù)據(jù)更加清晰和有序。

這樣，前端可以利用這棵樹來構(gòu)建下拉菜單等形式的界面，為用戶提供更好的體驗。

代碼如下：

public static void main(String[] args) {
    // todo 業(yè)務數(shù)據(jù)
    List<TreeVo> treeVoList = new ArrayList<>();

	// 處理父節(jié)點為空的情況
    treeVoList
            .stream()
            .filter(obj -> StringUtils.isBlank(obj.getPid()))
            .forEach(obj -> obj.setPid("-1"));

	// 根據(jù)父節(jié)點分組
    Map<String, List<TreeVo>> treeCatch = treeVoList.stream()
            .collect(Collectors.groupingBy(TreeVo::getPid));

    // 遞歸查詢子節(jié)點，寫入當前數(shù)據(jù)
    treeVoList = findChildren(treeCatch, "-1");

    // 打印
    for (TreeVo treeVo : treeVoList) {
        System.out.println(JacksonUtil.bean2json(treeVo));
    }
}

private static List<TreeVo> findChildren(Map<String, List<TreeVo>> treeCatch, String parentId) {
    List<TreeVo> children = treeCatch.getOrDefault(parentId, new ArrayList<>());
    for (TreeVo treeVo : children) {
        List<TreeVo> subChildren = findChildren(treeCatch, treeVo.getId());
        treeVo.getChildren().addAll(subChildren);
    }
    return children;
}



/**
 * 獲取場地樹RVO
 */
public class TreeVo {

    /**
     * ID
     */
    private String id;

    /**
     * 名稱
     */
    private String name;
    
    /**
     * 父ID
     */
    private String pid;

    /**
     * 子級
     */
    private transient List<TreeVo> children;
	
	// 此處省略get\set
}