js 通过id、pid遍历集合获得树结构

js 通过id、pid遍历集合获得树结构

原文章地址

  node=[{id:1,pid:0,name:"吉林省"},
		{id:2,pid:1,name:"长春市"},
		{id:3,pid:1,name:"松原市"},
		{id:4,pid:0,name:"辽宁省"},
		{id:5,pid:4,name:"沈阳市"},
		{id:6,pid:3,name:"扶余县"}
	   ]
/*
	通过定义map，key当前对象id，  value该对象
	遍历集合，得到对象顶级节点放到集合中返回
	不是顶级的就是当前对象的子节点，将对象放到该节点下，这一段比较抽象
*/

function toTree(node){
	node.forEach(function(it){
		delete it.children;
	})
	// 定义map/
	var map = {};
	// 这里可以重构数据类型，放回字段值
	node.forEach(function(it){
		map[it.id]=it;
	})
	
	// 定义返回集合
	var val=[];
	node.forEach(function(it){
		var parent = map[it.pid];
		if(parent){
			// 有数据说明不是顶级节点，将数据放到该 children 子节点下
			((parent.children) || (parent.children=[])).push(it);
		}else{
			// 没有数据说明是顶级节点放到val中
			val.push(it);
		}
	});
	return val;
}

console.log(JSON.stringify(toTree(node)))``