Javaで木構造のリストとフラットなリストの相互変換を実現する方法は何ですか？

木構造リストとフラットリストを相互変換する方法の1つは、再帰的に木構造を走査することです。

最初に、ツリーノードのクラスを定義して、ノードの値と子ノードのリストを含めます。

class TreeNode {
    private int value;
    private List<TreeNode> children;

    // 构造函数

    public TreeNode(int value) {
        this.value = value;
        this.children = new ArrayList<>();
    }

    // getter和setter方法

    public int getValue() {
        return value;
    }

    public void setValue(int value) {
        this.value = value;
    }

    public List<TreeNode> getChildren() {
        return children;
    }

    public void setChildren(List<TreeNode> children) {
        this.children = children;
    }

    // 添加子节点的方法

    public void addChild(TreeNode child) {
        this.children.add(child);
    }
}

次に、ツリーリストをフラットリストに変換する方法を実装します。

public class TreeListConverter {
    // 树形List转扁平List
    public static List<TreeNode> flattenList(List<TreeNode> treeList) {
        List<TreeNode> flatList = new ArrayList<>();
        for (TreeNode node : treeList) {
            flatList.add(node);
            flatList.addAll(flattenList(node.getChildren()));
        }
        return flatList;
    }
}

その後、フラットなリストをツリー形式のリストに変換する方法：

public class TreeListConverter {
    // 扁平List转树形List
    public static List<TreeNode> buildTreeList(List<TreeNode> flatList) {
        Map<Integer, TreeNode> nodeMap = new HashMap<>();
        List<TreeNode> treeList = new ArrayList<>();
        for (TreeNode node : flatList) {
            nodeMap.put(node.getValue(), node);
            if (node.getParentId() == null) {
                treeList.add(node);
            } else {
                TreeNode parent = nodeMap.get(node.getParentId());
                parent.addChild(node);
            }
        }
        return treeList;
    }
}

扁平リストをツリーリストに変換するためには、ノードクラスに親ノードへの参照を追加する必要があります。

例文：

public class Main {
    public static void main(String[] args) {
        List<TreeNode> treeList = new ArrayList<>();
        TreeNode node1 = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        TreeNode node3 = new TreeNode(3);
        TreeNode node4 = new TreeNode(4);
        TreeNode node5 = new TreeNode(5);
        TreeNode node6 = new TreeNode(6);
        node1.addChild(node2);
        node1.addChild(node3);
        node2.addChild(node4);
        node2.addChild(node5);
        node3.addChild(node6);
        treeList.add(node1);

        List<TreeNode> flatList = TreeListConverter.flattenList(treeList);
        System.out.println(flatList);

        List<TreeNode> treeList2 = TreeListConverter.buildTreeList(flatList);
        System.out.println(treeList2);
    }
}

結果出力：

[1, 2, 4, 5, 3, 6]
[1]