获取树对象属性 about js-challenges HOT 3 OPEN

function getCode(tree, cityName) {
  if (tree.name === cityName) {
    return tree.code || null;
  } else if (tree.children) {
    for (var i = 0; i < tree.children.length; i++) {
      var code = getCode(tree.children[i], cityName);
      if (code) {
        return code;
      }
    }
  }
  return null;
}

function getCode(tree, name) {
  let res = {}

  traversal(tree)
  return res

  function traversal(node) {
    if (!node) return

    if (node.name === name) {
      res = {...node}
      return
    }

    node.children && node.children.forEach(item => {
      traversal(item)
    })
  }
}

console.log(getCode(tree, '杭州市')) // { name: '杭州市', code: '0571' }