Java实现树的层序遍历

前言

最近居家办公，我们这周工作量比较少，于是我早早的进入了摸鱼时光，但是看到女朋友还在那辛苦的处理数据，一遍遍的重复工作。心想为何不写个程序，让女票也解放出来，一起快乐的摸鱼呢。

于是研究了下女友的工作，主要是从网站的一个画布流程上复制文本到Excel，并且将文本编号，指明下一个节点是哪个文本。

打开浏览器，F12 看了下响应数据，就是普通的 JSON 格式。 一共不到10个流程，也就懒得再去用程序爬取数据了，手动copy数据到文本文件，然后写个程序解析一下，因为原数据里有节点ID和子节点ID的集合，所以写起来不到半个小时就搞定了！

因为这里面编号的时候需要按照层级去编序号，所以需要按层级去给他遍历一下，这个时候就用到层序遍历了。

实现

Java 实现层序遍历一

节点对象

@Data
public class Node {
	
       /**
     * 节点ID
     */
    private String id;
    /**
     * 节点名称
     */
    private String name;
    /**
     * 子节点ID集合
     */
    private List<String> childIds;
    
}

这里通过原始数据已经获得了根节点ID和数据对象的集合，然后返回各层数据集合的一个集合就可以了。

层序遍历方法

    private static List<List<Node>> levelOrder(String rootId, List<Node> nodesList) {
        List<List<Node>> result = new ArrayList<>();

        if (rootId != null || nodesList == null || nodesList.isEmpty()) {
            return result;
        }

        Map<String, Node> nodeIdMap = nodesList
                .stream()
                .collect(Collectors.toMap(Node::getId, node -> node, (n1, n2) -> n1));

        Queue<String> queue = new LinkedList<>();
        queue.add(rootId);

        while (!queue.isEmpty()) {
            List<Node> nodeLevelList = new ArrayList<>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                String id = queue.poll();
                Node node = nodeIdMap.get(id);
                if (node == null) {
                    continue;
                }
                nodeLevelList.add(node);
                List<String> childNodeIds = node.getChildIds();
                if (childNodeIds != null && !childNodeIds.isEmpty()) {
                    queue.addAll(childNodeIds);
                }
            }
            result.add(nodeLevelList);
        }
        return result;
    }

Java 实现层序遍历二

如果你的对象结构是下面这样的，也就是对象里存放的是子节点而不是子节点ID

@Data
public class Node2 {
    /**
     * 节点ID
     */
    private String id;
    /**
     * 节点名称
     */
    private String name;
    /**
     * 子节点集合
     */
    private List<Node2> childNodes;

}

层级遍历方法如下：

    private static List<List<Node2>> levelOrder2(Node2 rootNode) {
        List<List<Node2>> result = new ArrayList<>();

        if (rootNode == null) {
            return result;
        }
        Queue<Node2> queue = new LinkedList<>();
        queue.add(rootNode);

        while (!queue.isEmpty()) {
            List<Node2> nodeLevelList = new ArrayList<>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                Node2 node = queue.poll();
                nodeLevelList.add(node);
                if (node == null) {
                    continue;
                }
                List<Node2> childNodes = node.getChildNodes();
                if (childNodes != null && !childNodes.isEmpty()) {
                    queue.addAll(childNodes);
                }
            }
            result.add(nodeLevelList);
        }
        return result;
    }

总结

这次帮女朋友写程序，也顺便帮自己熟悉了一遍树的层序遍历的实现，一举两得啊！

编程其实只是一种工具，并不是只能去完成自己的开发工作，也是可以用来帮自己或者身边的人解决日常工作中的重复性工作的。

如有疑问欢迎交流，不定期分享开发中自己造的一些小轮子，欢迎关注。