一天一大 leet(二叉树展开为链表)难度:中等-Day20200802

题目:

给定一个二叉树，原地将它展开为一个单链表。

例如，给定二叉树

将其展开为：

抛砖引玉

img

题意

将二叉树的所有右节点放到根节点右侧上
放置顺序：先右后左即某节点同时存在左右节点时优先将左侧节点追加右侧
前序遍历

思路

递归展开左侧所有节点依次追加
展开的节点本身还包含其自身的子节点，需要重新定义节点的子节点
- left -> null
- right -> 需要追加的下一个右节点

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {void} Do not return anything, modify root in-place instead.
 */
var flatten = function (root) {
  // 特殊情况：传入的二叉树为空直接返回
  if (!root || root.length === 0) {
    return root
  }

  let list = []
  helper(root)
  //  遍历所有节点
  for (let i = 0; i < list.length - 1; i++) {
    let node = list[i],
      nextNode = list[i + 1]
    node.left = null
    node.right = nextNode
  }

  // 收集节点
  function helper(node) {
    if (node !== null) {
      list.push(node)
      helper(node.left)
      helper(node.right)
    }
  }
}

其他解法

递归时已经遍历了所有节点
那么可以尝试不生成真实的 list，在遍历时就拼接二叉树
先 left 后 right
- 从根节点遍历时遇到 left 节点就将其遍历插入到原根节点 right 之前
- 根节点->leftNode-start-left->leftNode-end-right->rightNode

    1
   / 
  2   5
 /    
3   4   6
----------------------=>
    1
     
      2
     / 
    3   4
         
          5
           
            6
----------------------=>
1
 
  2
   
    3
     
      4
       
        5
         
          6

var flatten = function (root) {
  // 特殊情况：传入的二叉树为空直接返回
  if (!root || root.length === 0) {
    return root
  }
  function helper(node) {
    if (node !== null) {
      // 当前节点右侧节点
      let right = node.right
      // 将左节点放置到右节点  清除左节点，
      node.right = node.left
      node.left = null
//  遍历当前节点原左节点的右节点的根节点，使其余当前节点right节点连接
      let rightEnd = node
      while (rightEnd.right) {
        rightEnd = rightEnd.right
      }
      rightEnd.right = right
      // 右侧拼接的还有分支继续拼接
      helper(node.right)
    }
  }
  helper(root)
}