树转数组

思路

• 遍历树节点（广度优先遍历）
• 将树节点转为Array Item，push到数组
• 根据父子关系设置映射关系，并把子节点入队

interface IArrayItem {
  id: number
  name: string
  parentId: number
}

interface ITreeNode {
  id: number
  name: string
  children?: ITreeNode[]
}

function convert(root: ITreeNode): IArrayItem[] {
  const nodeToParent: Map<ITreeNode, ITreeNode> = new Map()
  const arr: IArrayItem[] = []

  // 广度优先遍历，queue
  const queue: ITreeNode[] = []
  // 根节点入队
  queue.unshift(root)
  
  while(queue.length > 0) {
    const curNode = queue.pop() // 出队
    if (curNode == null) break
    const { id, name, children = [] } = curNode
    // 创建数组item并push
    const parentNode = nodeToParent.get(curNode)
    const parentId = parentNode?.id || 0
    const item = { id, name, parentId }
    arr.push(item)
  
    // 子节点入队
    children.forEach(child => {
      nodeToParent.set(child, curNode)
      // 入队
      queue.unshift(child)
    })
  }
  return arr
}