LeetCode Flatten Binary Tree to Linked List 将二叉树展开成链表-蒲公英云

LeetCode Flatten Binary Tree to Linked List 将二叉树展开成链表

题目描述:

Flatten a binary tree to a fake “linked list” in pre-order traversal.
Here we use the right pointer in TreeNode as the next pointer in ListNode.

样例输入:

1
    / \    2   5
  / \   \  3   4   6

样例输出:

1
 \   2
   \     3
     \       4
       \         5
         \           6

解法一
使用递归的思想，首先使用DFS思路找到左子树的最左节点，然后将该最左节点的父节点和右子树断开，把该最左节点连接到父节点的右节点上，左节点置为null，再把原右子树连接到原左子树的最右节点上，再往上遍历

步骤:
首先找到最左节点3

1
    / \    2   5
  / \   \  3   4   6

断开父节点2和4的连接把3连接到2的右子节点再把4连接到3的右节点

1
    / \    2   5
    \   \      3   6
      \        4

再把1和右子树5断开把左子树2连接到1的右节点上右子树5连接到左子树的最右节点4的右节点上

1
 \   2
   \     3
     \       4
       \         5
         \           6

代码

public void flattenRecursion(TreeNode root) {
    if (root == null) {
        return;
    }
    /*使用DFS思路找到最左子节点*/
    if (root.left != null) {
        flattenRecursion(root.left);
    }
    if (root.right != null) {
        flattenRecursion(root.right);
    }
    /*把其父节点和右子节点断开，将原左子结点连上父节点的右子节点上*/
    TreeNode tmp = root.right;
    root.right = root.left;
    root.left = null;
    /*然后再把原右子节点连到新右子节点的右子节点上*/
    while (root.right != null) {
        root = root.right;
    }
    root.right = tmp;
}

解法2
【递归能实现的循环就能实现┐(‘～`；)┌ 】首先判断当前节点的左子树是否为空，如果左子树非空的话，就断开当前节点和右子树，把当前节点的右指针指向左子树，再把原右子树连接到原左子树的最右节点
步骤描述：

1
    / \    2   5
  / \   \  3   4   6

首先根节点1的左子树2非空断开1和5的连接把1的右指针指向2 再把右子树5连接到2的最右节点4的右节点上

1
      \        2
      / \      3   4
          \            5
            \              6

再继续判断以2为根的树 2的左节点3非空继续进行上述操作

1
 \   2
   \     3
     \       4
       \         5
         \           6

代码实现

public void flattenLoop(TreeNode root) {
   public void flattenLoop(TreeNode root) {
    TreeNode cur = root;
    while (cur != null) {
        //遍历节点 如果左节点不为空的话
        if (cur.left != null) {
            //找到左子树的最右节点 目的是把原右子树连接到该节点的右节点上
            TreeNode p = cur.left;
            while (p.right != null) {
                p = p.right;
            }
            //断开当前节点和右子树 把左子树连接到当前节点的右节点 把原右子树连接到原左子树的最右节点的右子树上
            p.right = cur.right;
            cur.right = cur.left;
            cur.left = null;
        }
        //当前节点为右子树的根节点 即原左子树的根节点
        cur = cur.right;
    }
}

解法三
使用层序遍历来实现需要借助辅助空间来实现

public void flattenPreOrder(TreeNode root) {
    if(root == null){
         return;
     }
     Stack<TreeNode> stack = new Stack<TreeNode>();
     stack.push(root);
     while(!stack.empty()){
         TreeNode tmp = stack.pop();
         if(tmp.left!=null){
             TreeNode l = tmp.left;
             while(l.right!=null){
                 l = l.right;
             }
             TreeNode r = tmp.right;
             tmp.right = tmp.left;
             tmp.left = null;
             l.right = r;
         }
         if(tmp.right!=null){
             stack.push(tmp.right);
         }
     }
 }