List数据快速转换为Tree 树结构

IT洋少

已于 2024-05-15 17:21:33 修改

阅读量1.1k

点赞数

分类专栏： java 文章标签： java

于 2021-07-05 17:01:52 首次发布

本文链接：https://blog.youkuaiyun.com/qq_28087959/article/details/118493733

版权

java 专栏收录该内容

4 篇文章

订阅专栏

将List数据快速转换为Tree 树结构

看了现在很多网友写的，发现他们写的代码逻辑都会对数据进行多次循环，这样性能我个人认为不是很好，所以我这边贴出我的代码，这边只用到一次循环，供大家参考，和一起学习，希望大家一起提出更好的想法。

/**
树结构实体 大部分基本上都是这样 需要的自行改造
*/
public class Node {
	    public Node(String id, String pId, String string) {
			this.id = id;
			pid = pId;
			name = string;
			this.children = new ArrayList<Node>();
			
		}
		/**
	     * id
	     */
	    private String id;
	    /**
	     * 父节点id
	     */
	    private String pid;
	    /**
	     * 名称
	     */
	    private String name;
	    /**
	     * 子节点
	     */
	    private List<Node> children;
}

下面我们来展示真正转换的逻辑代码代码

public List<Node> buildTree(List<Node> nodes){
    	Map<String, Node> nodeMap = new HashMap<String, BuildTree.Node>();
    	List<Node> roots = new ArrayList<Node>();
    	for (Node node : nodes) {
			String id = node.getId();
			Node n = nodeMap.get(id);
			if(n == null){
				n = node;
				nodeMap.put(id, node);
			}else {
				String pid = n.getPid();
				if(pid.equals("-1")){//代表是我们预先存的节点 替换真实数据
					copyProps(n,node);//复制数据逻辑
				}
			}
			String pid = node.getPid();
			Node pn = nodeMap.get(pid);
			if(pn == null){//父节点不存在 先设置假数据 存在
				pn = new Node(pid, "-1", "");
				if(pid.equals("0")){
					roots.add(pn);
				}
				nodeMap.put(pid, pn);
			}
			pn.children.add(n);
		}
       return roots.get(0).getChildren();//返回真正顶级节点
    }