面试复盘-树相关

//输入：
const list = [
    { key: 1, data: 'A', parentKey: 0 },
    { key: 2, data: 'B', parentKey: 0 },
    { key: 3, data: 'C', parentKey: 1 },
    { key: 4, data: 'D', parentKey: 1 },
    { key: 5, data: 'E', parentKey: 2 },
    { key: 6, data: 'F', parentKey: 3 },
    { key: 7, data: 'G', parentKey: 2 },
    { key: 8, data: 'H', parentKey: 4 },
];
//输出：
// [
//   {
//     key: 1,
//     data: 'A',
//     parentKey: 0,
//     children: [
//       { key: 3, data: 'C', parentKey: 1, children: [{ key: 6, data: 'F', parentKey: 3 }] },
//       { key: 4, data: 'D', parentKey: 1, children: [{ key: 8, data: 'H', parentKey: 4 }] },
//     ],
//   },
//   {
//     key: 2,
//     data: 'B',
//     parentKey: 0,
//     children: [
//       { key: 5, data: 'E', parentKey: 2 },
//       { key: 7, data: 'G', parentKey: 2 },
//     ],
//   },
// ];

const result = convert(list);
function convert(list) {
    // your code
    let childrens = new Array(list.length + 1);
    let res = [];
    for(let i = 0; i <= list.length; i++){
        childrens[i] = [];
    }
    
    for (let item of list) {
        childrens[item.parentKey].push(item);
    }
    //初始化,parentKey = 0
    for(let item of childrens[0]){
        res.push(item);
    }
    const dfs = (item) => {
        if(childrens[item.key].length === 0) return;
        item.children = [];
        for(let ch of childrens[item.key]){
            item.children.push(ch);
            dfs(ch);
        }
    };
    for(let item of res){
        dfs(item);
    }
    return res;
}