构造树结构,已知若干个子节点的id,需要将子节点和它的父节点的数据构造成一棵树,没有的子节点不展示。

public static void main(String[] args) {
    List<Concept> concepts = new ArrayList<>();
    concepts.add(new Concept(0, 0, "顶级"));
    concepts.add(new Concept(1, 0, "顶级1"));
    concepts.add(new Concept(2, 1, "第一级1"));
    concepts.add(new Concept(3, 1, "第一级2"));
    concepts.add(new Concept(4, 2, "第二级1"));
    concepts.add(new Concept(5, 2, "第二级2"));
    concepts.add(new Concept(6, 4, "第三级1"));
    concepts.add(new Concept(7, 4, "第三级2"));
    concepts.add(new Concept(8, 6, "第四级1"));
    List<Integer> targetIds = new ArrayList<>();
    targetIds.add(3);
    targetIds.add(6);
    targetIds.add(8);
    Concept root = buildFocusedTree(concepts, targetIds);
    System.out.println(root);
}
public static Concept buildFocusedTree(List<Concept> concepts, List<Integer> targetIds) {
    Map<Integer, Concept> conceptMap = new HashMap<>();
    Map<Integer, Concept> allParentsMap = new HashMap<>();
    for (Concept concept : concepts) {
        conceptMap.put(concept.id, concept);
    }
    for (int targetId : targetIds) {
        Concept current = conceptMap.get(targetId);
        while (current != null && current.id != 0) {
            allParentsMap.put(current.id, current);
            current = conceptMap.get(current.parentId);
        }
        if (current != null && current.id == 0) {
            allParentsMap.put(current.id, current);
        }
    }
    for (Concept concept : allParentsMap.values()) {
        Concept parent = conceptMap.get(concept.parentId);
        if (parent != null && allParentsMap.containsKey(parent.id) && !parent.children.contains(concept)
            && (concept.getId() != 0)) {
            parent.children.add(concept);
        }
    }
    return conceptMap.get(0);
}
  • 10
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个使用Java实现的算法,可以根据父节点查询所有子节点并拼一棵树: ```java import java.util.*; public class TreeNode { private int treeId; private int parentId; private List<TreeNode> children; public TreeNode(int treeId, int parentId) { this.treeId = treeId; this.parentId = parentId; this.children = new ArrayList<>(); } public int getTreeId() { return treeId; } public int getParentId() { return parentId; } public List<TreeNode> getChildren() { return children; } public void addChild(TreeNode child) { children.add(child); } public static List<TreeNode> buildTree(List<TreeNode> nodes) { Map<Integer, TreeNode> nodeMap = new HashMap<>(); for (TreeNode node : nodes) { nodeMap.put(node.getTreeId(), node); } List<TreeNode> rootNodes = new ArrayList<>(); for (TreeNode node : nodes) { if (node.getParentId() == 0) { rootNodes.add(node); } else { TreeNode parent = nodeMap.get(node.getParentId()); if (parent != null) { parent.addChild(node); } } } return rootNodes; } public static void main(String[] args) { // 构造节点数据 List<TreeNode> nodes = new ArrayList<>(); nodes.add(new TreeNode(1, 0)); nodes.add(new TreeNode(2, 1)); nodes.add(new TreeNode(3, 1)); nodes.add(new TreeNode(4, 2)); nodes.add(new TreeNode(5, 4)); nodes.add(new TreeNode(6, 3)); nodes.add(new TreeNode(7, 3)); nodes.add(new TreeNode(8, 6)); nodes.add(new TreeNode(9, 7)); // 构建 List<TreeNode> rootNodes = TreeNode.buildTree(nodes); // 打印 for (TreeNode rootNode : rootNodes) { printTree(rootNode, 0); } } private static void printTree(TreeNode node, int level) { StringBuilder sb = new StringBuilder(); for (int i = 0; i < level; i++) { sb.append(" "); } sb.append(node.getTreeId()); System.out.println(sb.toString()); for (TreeNode child : node.getChildren()) { printTree(child, level + 1); } } } ``` 该算法的核心是 `buildTree` 方法,它接受一个节点列表作为参数,返回一个根节点列表。该方法首先遍历节点列表,将每个节点存储在一个 `Map` 中,以便可以根据节点的 `treeId` 快速查找节点。然后遍历节点列表,对于每个节点,如果 `parentId` 为 0,则将其添加到根节点列表中;否则,从 `Map` 中获取其父节点,如果父节点存在,则将该节点添加为父节点子节点。最后返回根节点列表即可。 在 `main` 方法中,我们构造了一个节点列表,并使用 `buildTree` 方法构建一棵树。然后调用 `printTree` 方法打印的结构。该方法接受一个节点和一个层级作为参数,递归地打印的结构。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值