【LeetCode從零單排】No 114 Flatten Binary Tree to Linked List

李博Garvin發表於2015-03-16
LeetCode

題目

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

         1
        / \
       2   5
      / \   \
     3   4   6

The flattened tree should look like:
   1
    \
     2
      \
       3
        \
         4
          \
           5
            \
             6
解題思路：利用遞迴找到倒數第一個父節點，記錄下它的右節點，將左邊的移到右邊，然後再把之前標記的右節點連線上。

程式碼

public class Solution {
    public void flatten(TreeNode root) {
        if(root==null) return;
        flatten(root.left);
        flatten(root.right);
        TreeNode temp=root.right;
        if(root.left!=null){
            root.right=root.left;
            root.left=null;
            
        
        while(root.right != null){
           root=root.right;
        }
        root.right=temp;
            
        }
        
    }
}




/********************************

* 本文來自部落格  “李博Garvin“

* 轉載請標明出處:http://blog.csdn.net/buptgshengod

******************************************/



相關文章