dave-wind
commented
1 year ago 算法
首先介绍 DFS 深度优先搜索
DFS（深度优先搜索）算法，用于找到从起点到目标节点的最短路径。 ,* 给定一个起始节点，
这将返回树中具有目标值的起始节点下方的节点（如果不存在，则返回 null） ,
* 在 O(n) 中运行，其中 n 是树中的节点数，或 O(b^d)，其中 b 是分支因子，d 是深度。
const dfs = function (start, target) {
console.log("Visiting Node " + start.value);
if (start.value === target) {
// We have found the goal node we we're searching for
console.log("Found the node we're looking for!");
return start;
}
// Recurse with all children
for (var i = 0; i < start.children.length; i++) {
var result = dfs(start.children[i], target);
if (result != null) {
// We've found the goal node while going down that child
return result;
}
}
// We've gone through all children and not found the goal node
console.log("Went through all children of " + start.value + ", returning to it's parent.");
return null;
};
解: 性能很差 因为每次都要找到底部;
// 根据DFS原理 就好比俩点找最短路径, 都是一条路走到底不行再换下一条;
// 所以判断子元素parentId == 目标id的时候 就一直递归下去;直到不等于;但是 性能不好 不是 O(n)
function convert(source, parentId = 0){
let trees = [];
for (let item of source) {
// 递归条件 也是基线条件 当不满足就直接 不会递归了
if(item.parentId === parentId) {
let children = convert(source, item['id']);
if(children.length) {
item.children = children
}
trees.push(item);
}
}
return trees;
}
递归法: O(n^2)
function convert(arr, result = []) {
for (let index = 0; index < arr.length; index++) {
// 当前每一个对象
let current = arr[index]
if (current.parentId == 0) { // 为0 直接push
result.push(current);
} else {
const findParent = arr.find(item => current.parentId == item.id); // 去找父级
if (findParent) {
findParent.children = findParent.children || [];
findParent.children.push(current);
// 第一次循环结束 继续下一次循环
convert2(findParent.children, result)
}
}
}
return result;
}
最佳解法: O(n) + O(n) ~= O(n)
function createMap(arr) {
let obj = {}
arr.forEach((item) => {
obj[item.id] = item; // 注意这里 不能用 对象解构 扩展符 因为就是为了 引用指向同一个
obj[item.id].children = []
})
return obj
}
function createTree(arr, map) {
let result = []
arr.forEach((item) => {
// 为0 做根元素
if (item.parentId == 0) {
result.push(item);
} else {
// 通过 parentid 去找 父 id; 这里利用的就是引用一样; 无论多深层级 都可以找到向上的父级
const findParent = map[item.parentId];
if (findParent) {
findParent.children.push(item)
}
}
})
return result;
}
function convert(arr) {
const map = createMap(arr);
return createTree(arr, map)
}
数组转tree