java树结构筛选第一层级

Java 中筛选树结构的第一层级
在Java中处理树形结构时，通常会遇到需要提取特定层级节点的需求。对于第一层级节点的筛选，可以通过定义一个方法来获取根节点列表，并确保这些节点确实位于树的最顶层[^1]。
定义实体类
为了更好地操作树形数据，在此之前应该先设计好相应的实体类，假设有一个TreeNode类代表树中的每一个节点：
public class TreeNode {
private Long id;
private Long parentId; // 如果是顶级节点，则parentId为null或指定特殊值如0L
private String name;
@TableField(exist = false)
private List<TreeNode> children;

// Getters and Setters...
}
登录后复制
获取最顶层的数据
接下来创建一个函数用于收集所有的顶级节点。这一步骤通过判断parentId是否等于预设的顶级标志（比如null或者某个固定ID）来进行过滤[^4]:
private static final Long TOP_LEVEL_PARENT_ID = null;

public List<TreeNode> getTopLevelNodes(List<TreeNode> allNodes){
return allNodes.stream()
.filter(node -> Objects.equals(TOP_LEVEL_PARENT_ID, node.getParentId()))
.collect(Collectors.toList());
}
登录后复制
上述代码片段展示了如何利用流式API(Stream)配合条件语句从给定的全部节点集中挑选出那些属于最高级别的实例。注意这里的比较采用了Objects.equals()而不是简单的==运算符，这是为了避免当TOP_LEVEL_PARENT_ID设置成null时可能出现的问题[^2]。