LeetCode 156 Binary Tree Upside Down 上下翻轉(zhuǎn)二叉樹(shù)

Loong_T 發(fā)布于2019-08-16 13:28 / 3412人閱讀

摘要：翻轉(zhuǎn)以后如下解題思路翻轉(zhuǎn)的形式一開(kāi)始不是很清楚，但是里面的高票答案給了一個(gè)很好的解釋。看例子，樹(shù)的左邊最深的底層是，是新的。對(duì)于每個(gè)，將鏈接右孩子的指針去掉，將變?yōu)楫?dāng)前左孩子的，成為左孩子的。遞歸的寫(xiě)法遞歸調(diào)用得到新的，并且沿途改變結(jié)構(gòu)。

LeetCode 156 Binary Tree Upside Down

Given a binary tree where all the right nodes are either leaf nodes with a sibling (a left node that shares the same parent node) or empty, flip it upside down and turn it into a tree where the original right nodes turned into left leaf nodes. Return the new root.

Example:

Input: [1,2,3,4,5]

Output: return the root of the binary tree [4,5,2,#,#,3,1]
翻轉(zhuǎn)以后如下：

解題思路：
翻轉(zhuǎn)的形式一開(kāi)始不是很清楚，但是discuss里面的高票答案給了一個(gè)很好的解釋。看例子，樹(shù)的左邊最深的底層是4，4是新的root。對(duì)于每個(gè)root node，將鏈接右孩子的指針去掉，將root node變?yōu)楫?dāng)前左孩子的left node，root node成為左孩子的right node。

    1
   /  x
  2 -- 3
 /  x
4 -- 5
^
new root

遞歸的寫(xiě)法：

    public TreeNode upsideDownBinaryTree(TreeNode root) {
        if (root == null || root.left == null) {
            return root;
        }
        //遞歸調(diào)用得到新的root，并且沿途改變結(jié)構(gòu)。
        TreeNode newRoot = upsideDownBinaryTree(root.left);
        root.left.left = root.right;
        root.left.right = root;
        //千萬(wàn)記得將root node 的左右兩邊設(shè)為null
        root.left = null;
        root.right = null;
        return newRoot;
    }

遍歷的解法
遍歷的解法需要四個(gè)指針，如圖所示，每次先update next，然后對(duì)swap上一個(gè)node的右孩子和這個(gè)node的左孩子，所以每次我們需要一個(gè)temp來(lái)記錄上一個(gè)node的右邊孩子。

   prev -> 1
          /  x
 curr -> 2 -- 3  <-temp
        /  x
next-> 4 -- 5
      ^
    new root

代碼如下

    public TreeNode upsideDownBinaryTree(TreeNode root) {
        //iterative
        TreeNode curr = root;
        TreeNode prev = null;
        TreeNode next = null;
        TreeNode temp = null;
        while(curr != null) {
            next = curr.left;
            //swap nodes, we need to keep a temp to track the right node
            curr.left = temp;
            temp = curr.right;
            curr.right = prev;
            
            prev = curr;
            curr = next;
        }
        return prev;