阿里面试题 树的深度优先遍历

前几天内推阿里的前端，自己本来就是菜鸟，面试官让做一道编程题，20分钟。我看到编程题，树的深度优先遍历，这名字让我想到了图的深度优先遍历，心想完了，完了，图的算法我根本不熟悉啊，于是心态就崩了，最后面试也失败了。

后来自己静下心来看这道题，发现并不难，题目如下：

//请使用 JavaScript 编写一个树的深度优先遍历函数（节点最深的最先访问到，依次类推），满足以下测试用例：

// 假设树的结构如下

const tree = [
  {
    id: 1,
    name: '张三',
    children: [
        {
            id: 2,
            name: '李四',
            children: [
                {
                    id: 5,
                    name: '张五'
                }
            ]
        }
    ]
  },
  {
    id: 6,
    name: '玛丽'
  }
]


// 测试用例：

// 1. 生成一颗新树，将所有节点的id，加1
console.log(treeMap(tree, node => {
    let newNode = { ...node }
    newNode.id = node.id + 1
    return newNode
}))

// 打印的新树，应该与tree的结构一致，只是每个id自增1，老的tree，应该没有任何改动

// 2. 打印每个节点的id
treeMap(tree, node => {
    console.log(node.id)
    return node
});


// 应打印顺序应该是： 5，2，1，6


// 3. 对于非法输入，应直接返回第一个入参
console.log(treeMap(null)) // 输出null
console.log(treeMap(tree, true/*不是函数*/)) //输出tree

我的解答如下

function treeMap(tree, f){
    const newtree = [];
    if (!(f instanceof Function)){
        return tree;
    }
    if (tree instanceof Array){
        for (let item of tree){
            const obj = Object.assign({}, item) //这里要注意克隆一下节点，不然会影响老树的结构
            if (item.hasOwnProperty("children")){
                obj.children = treeMap(item.children,f)
            }
            newtree.push(f(obj))
        }
    }
    return newtree
}