树的应用方法


    @Data
    @AllArgsConstructor
    @NoArgsConstructor
    @Builder
    public static class Node {
        private String name;
        private String pName;
        private List<Node> node;
    }


    public static List<Node> getNodeList() {
        return Lists.newArrayList(new Node("1", "", new ArrayList<>()),
                new Node("1-2", "1", new ArrayList<>()),
                new Node("1-3", "1", new ArrayList<>()),
                new Node("1-4", "1", new ArrayList<>()),
                new Node("2-5", "1-2", new ArrayList<>()),
                new Node("2-6", "1-2", new ArrayList<>()),
                new Node("3-7", "1-3", new ArrayList<>()));
    }

    // 构建一颗树
    public static Node getNodeTree(List<Node> list) {
        Map<String, List<Node>> childMap = list.stream()
                .collect(Collectors.groupingBy(Node::getPName));
        list.stream().forEach(node -> node.setNode(childMap.getOrDefault(node.getName(), new ArrayList<>())));
        return list.stream().filter(node -> StringUtils.isBlank(node.getPName())).findAny().orElse(null);
    }


    // 获取子树
    public static Node getSubNodeTree(Node node, String nodeName) {
        if (StringUtils.equals(node.getName(), nodeName)) {
            return node;
        }

        for (Node child : node.getNode()) {
            Node n = getSubNodeTree(child, nodeName);
            if (n != null) {
                return n;
            }
        }
        return null;
    }

    // 获取子树的平铺的节点
    public static List<Node> getSubNodeList(Node node, String nodeName) {
        Node node1 = getSubNodeTree(node, nodeName);
        if (node1 == null) {
            return null;
        }
        return getTreeToNodeList(node1);
    }

    // 平铺树,转化为列表
    public static List<Node> getTreeToNodeList(Node node) {
        List<Node> list = Lists.newArrayList(node);

        if (CollectionUtils.isEmpty(node.getNode())) {
            return list;
        }

        for (Node child : node.getNode()) {
            list.addAll(getTreeToNodeList(child));
        }

        return list;
    }


    // 复制树
    public static Node copyNodeTree(Node node) {
        Node newNode = Node.builder().name("new" + node.getName()).node(new ArrayList<>()).build();

        if(CollectionUtils.isEmpty(node.getNode())){
            return node;
        }

        for (Node child : node.getNode()) {
            if (child != null) {
                newNode.getNode().add(copyNodeTree(child));
            }
        }

        return newNode;
    }

     // 穷举所有根节点到叶子节点的的树名字
    public static List<List<Node>> getPathNode(Node node, List<Node> nodes) {
        List<Node> nodeList = new ArrayList<>(nodes);
        nodeList.add(node);
        List<List<Node>> pathList = new ArrayList<>();
        if (CollectionUtils.isEmpty(node.getNode())) {
            pathList.add(nodeList);
            return pathList;
        }
        for (Node child : node.getNode()) {
            pathList.addAll(getPathNode(child, nodeList));
        }
        return pathList;
    }

    public static void main(String[] args) {
        // 构建node-tree
        Node tree = getNodeTree(getNodeList());
        System.out.println("获取node-tree" + tree);

        // 获取子树-有值
        Node n1 = getSubNodeTree(tree, "3-7");
        System.out.println("获取子树3-7" + n1);

        // 获取子树-空的
        Node n2 = getSubNodeTree(tree, "7");
        System.out.println("获取子树7" + n2);

        // 获取平铺的子树列表
        List<Node> list = getSubNodeList(tree, "1-2");
        System.out.println("获取子树1-2的列表 " + list);

        //复刻一颗新树
        Node n3 = copyNodeTree(tree);
        System.out.println("新树" + n3);


        List<List<Node>> n4 = getPathNode(tree,new ArrayList<>());
        System.out.println("所有路径的树列表" + n4);

    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值