快捷導(dǎo)航

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

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

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

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

java遞歸獲取樹

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

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

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

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

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

代碼如下：

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

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

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

    // 遞歸查詢子節(jié)點(diǎn)，寫入當(dāng)前數(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
}