前言
最近居家办公,我们这周工作量比较少,于是我早早的进入了摸鱼时光,但是看到女朋友还在那辛苦的处理数据,一遍遍的重复工作。心想为何不写个程序,让女票也解放出来,一起快乐的摸鱼呢。
于是研究了下女友的工作,主要是从网站的一个画布流程上复制文本到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;
}
总结
这次帮女朋友写程序,也顺便帮自己熟悉了一遍树的层序遍历的实现,一举两得啊!
编程其实只是一种工具,并不是只能去完成自己的开发工作,也是可以用来帮自己或者身边的人解决日常工作中的重复性工作的。
如有疑问欢迎交流,不定期分享开发中自己造的一些小轮子,欢迎关注。