java 遞歸查詢所有子節(jié)點id的方法實現

更新時間：2024年03月05日 11:50:04 作者：654dcecc9889d

在多層次的數據結構中,經常需要查詢一個節(jié)點下的所有子節(jié)點,本文主要介紹了java 遞歸查詢所有子節(jié)點id的方法實現,具有一定的參考價值,感興趣的可以了解一下

數據模型

首先，定義一個簡單的節(jié)點類Node，該類包含節(jié)點ID、父節(jié)點ID和子節(jié)點列表作為其屬性。

public class Node {
    private int id;
    private int parentId;
    private List<Node> children;

    // 構造函數、Getter和Setter省略
}

遞歸查詢實現

遞歸方法的核心思想是從根節(jié)點開始，遍歷其子節(jié)點，并對每個子節(jié)點重復此過程，直到到達葉子節(jié)點。下面是這一遞歸查詢邏輯的實現：

import java.util.ArrayList;
import java.util.List;

public class TreeUtils {

    /**
     * 遞歸查詢一個節(jié)點下的所有子節(jié)點ID
     * @param rootNode 起始節(jié)點
     * @return 所有子節(jié)點的ID列表
     */
    public static List<Integer> findAllChildNodeIds(Node rootNode) {
        List<Integer> childNodeIds = new ArrayList<>();
        addAllChildNodeIds(rootNode, childNodeIds);
        return childNodeIds;
    }

    private static void addAllChildNodeIds(Node node, List<Integer> childNodeIds) {
        if (node == null) {
            return;
        }
        List<Node> children = node.getChildren();
        if (children != null) {
            for (Node child : children) {
                childNodeIds.add(child.getId());
                addAllChildNodeIds(child, childNodeIds); // 遞歸調用
            }
        }
    }
}

在這個實現中，findAllChildNodeIds方法是公開調用的入口，它初始化一個空的ID列表，然后調用私有方法addAllChildNodeIds來填充這個列表。遞歸的停止條件是當前節(jié)點為null或其子節(jié)點列表為空。

使用示例

以下是如何使用上述工具類的一個簡單示例：

public class Main {
    public static void main(String[] args) {
        // 構造節(jié)點關系
        Node root = new Node();
        root.setId(1);
        
        Node child1 = new Node();
        child1.setId(2);
        
        Node child2 = new Node();
        child2.setId(3);
        
        root.setChildren(Arrays.asList(child1, child2));
        
        // 查詢所有子節(jié)點ID
        List<Integer> allChildNodeIds = TreeUtils.findAllChildNodeIds(root);
        
        // 輸出結果
        System.out.println("All Child Node IDs: " + allChildNodeIds);
    }
}

以上示例首先創(chuàng)建了一個簡單的樹形結構，然后使用TreeUtils.findAllChildNodeIds方法查詢根節(jié)點下的所有子節(jié)點ID，并打印這些ID。