import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;
 
// 樹節(jié)點類，用于存儲樹中的每個元素
class TreeNode<T> {
    private T data; // 節(jié)點存儲的數(shù)據(jù)
    private List<TreeNode<T>> children = new ArrayList<>(); // 子節(jié)點列表
 
    // 構造函數(shù)，初始化節(jié)點數(shù)據(jù)
    public TreeNode(T data) {
        this.data = data;
    }
 
    // 添加子節(jié)點
    public void addChild(TreeNode<T> child) {
        children.add(child);
    }
 
    // 獲取當前節(jié)點的數(shù)據(jù)
    public T getData() {
        return data;
    }
 
    // 獲取當前節(jié)點的所有子節(jié)點
    public List<TreeNode<T>> getChildren() {
        return children;
    }
}
 
// 樹構建器類，負責從列表構建樹形結構
class TreeBuilder<T> {
    // buildTree方法，構建樹形結構
    public TreeNode<T> buildTree(List<T> items, Function<T, String> idExtractor, Function<T, String> parentIdExtractor) {
        // 創(chuàng)建一個映射，將每個節(jié)點的ID映射到對應的TreeNode對象
        Map<String, TreeNode<T>> nodeMap = items.stream()
                .map(item -> new TreeNode<>(item)) // 將每個元素轉(zhuǎn)換為TreeNode
                .collect(Collectors.toMap(node -> idExtractor.apply(node.getData()), Function.identity()));
 
        TreeNode<T> root = null; // 用于存儲根節(jié)點
        
        // 遍歷每個元素，建立父子關系
        for (T item : items) {
            String id = idExtractor.apply(item); // 獲取當前節(jié)點的ID
            String parentId = parentIdExtractor.apply(item); // 獲取父節(jié)點的ID
            TreeNode<T> node = nodeMap.get(id); // 獲取當前節(jié)點
            
            if (parentId == null) {
                root = node; // 假設只有一個根節(jié)點
            } else {
                TreeNode<T> parentNode = nodeMap.get(parentId); // 獲取父節(jié)點
                if (parentNode != null) {
                    parentNode.addChild(node); // 將當前節(jié)點添加到父節(jié)點的子節(jié)點列表中
                }
            }
        }
        
        return root; // 返回構建的樹的根節(jié)點
    }
}

import java.util.Arrays;
import java.util.List;
 
// 實體類，用于表示樹節(jié)點的數(shù)據(jù)
class Entity {
    private String id;         // 節(jié)點ID
    private String parentId;   // 父節(jié)點ID
    private String name;       // 節(jié)點名稱
 
    // 構造函數(shù)
    public Entity(String id, String parentId, String name) {
        this.id = id;
        this.parentId = parentId;
        this.name = name;
    }
 
    // 獲取節(jié)點ID
    public String getId() {
        return id;
    }
 
    // 獲取父節(jié)點ID
    public String getParentId() {
        return parentId;
    }
 
    // 獲取節(jié)點名稱
    public String getName() {
        return name;
    }
}
 
// 主程序
public class Main {
    public static void main(String[] args) {
        // 創(chuàng)建實體列表，模擬樹形結構
        List<Entity> entities = Arrays.asList(
            new Entity("1", null, "Root"),         // 根節(jié)點
            new Entity("2", "1", "Child 1"),       // Child 1, 父節(jié)點為Root
            new Entity("3", "1", "Child 2"),       // Child 2, 父節(jié)點為Root
            new Entity("4", "2", "Grandchild 1")   // Grandchild 1, 父節(jié)點為Child 1
        );
 
        // 創(chuàng)建樹構建器實例
        TreeBuilder<Entity> treeBuilder = new TreeBuilder<>();
        
        // 構建樹，傳入實體列表及提取ID和父ID的函數(shù)
        TreeNode<Entity> tree = treeBuilder.buildTree(
            entities,
            Entity::getId,        // 提取ID
            Entity::getParentId   // 提取父ID
        );
 
        // 打印樹結構
        printTree(tree, 0);
    }
 
    // 遞歸打印樹結構的輔助方法
    private static void printTree(TreeNode<Entity> node, int level) {
        if (node == null) return; // 如果節(jié)點為空，直接返回
 
        // 打印當前節(jié)點的名稱，縮進顯示層級
        System.out.println("  ".repeat(level) + node.getData().getName());
        
        // 遞歸打印所有子節(jié)點
        for (TreeNode<Entity> child : node.getChildren()) {
            printTree(child, level + 1);
        }
    }
}