Java构建树形结构,根据父级获取所有子级,以及子级获取父级
时间: 2023-05-24 09:01:46 浏览: 1198
java树形结构递归查询
您可以使用递归算法实现树形结构的父子级查询。首先,您需要根据节点的ID和parentID来构建树形结构,然后根据节点的ID查询其所有子级,或者根据子节点的ID查询其父节点。以下是Java代码示例:
// 树形结构节点类
public class TreeNode {
private String id;
private String parentId;
private List<TreeNode> children;
public TreeNode(String id, String parentId) {
this.id = id;
this.parentId = parentId;
children = new ArrayList<>();
}
// 添加子节点
public void addChild(TreeNode node) {
children.add(node);
}
// 获取所有子节点
public List<TreeNode> getChildren() {
return children;
}
// 获取父节点
public TreeNode getParent() {
// 查询父节点
return null;
}
}
// 构建树形结构
public static TreeNode buildTree(List<TreeNode> nodeList) {
Map<String, TreeNode> map = new HashMap<>();
// 将所有节点添加到map中
for (TreeNode node : nodeList) {
map.put(node.getId(), node);
}
// 构建树形结构
for (TreeNode node : nodeList) {
if (!node.getParentId().equals("0")) {
map.get(node.getParentId()).addChild(node);
}
}
// 返回根节点
return map.get("0");
}
// 根据节点ID获取所有子节点
public static List<TreeNode> getChildNodeList(TreeNode node) {
List<TreeNode> childNodeList = new ArrayList<>();
if (node != null && !node.getChildren().isEmpty()) {
for (TreeNode childNode : node.getChildren()) {
childNodeList.addAll(getChildNodeList(childNode));
}
childNodeList.addAll(node.getChildren());
}
return childNodeList;
}
// 根据子节点ID获取父节点
public static TreeNode getParentNode(TreeNode rootNode, String childId) {
if (rootNode != null) {
if (rootNode.getId().equals(childId)) {
return null;
}
for (TreeNode childNode : rootNode.getChildren()) {
if (childNode.getId().equals(childId)) {
return rootNode;
}
TreeNode parentNode = getParentNode(childNode, childId);
if (parentNode != null) {
return parentNode;
}
}
}
return null;
}
// 使用示例
List<TreeNode> nodeList = new ArrayList<>();
nodeList.add(new TreeNode("0", "0"));
nodeList.add(new TreeNode("1", "0"));
nodeList.add(new TreeNode("2", "1"));
nodeList.add(new TreeNode("3", "2"));
nodeList.add(new TreeNode("4", "2"));
TreeNode rootNode = buildTree(nodeList);
List<TreeNode> childNodeList = getChildNodeList(rootNode); // 获取所有子节点
TreeNode parentNode = getParentNode(rootNode, "4"); // 获取父节点
阅读全文