Leetcode_114_Flatten Binary Tree to Linked List__LeetCode

本文是在學習中的總結，歡迎轉載但請註明出處：http://blog.csdn.net/pistolove/article/details/42744919

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

思路：

（1）題意為將給定的二叉樹轉化為“只有右孩子節點”的鏈表（樹）。

（2）由上圖可知，如果右子樹不為空白，則右子樹最後肯定為左子樹最有一個靠右的孩子節點的右子樹，而左子樹最後成為整棵樹的右子樹。這樣，首先判斷左子樹是否為空白，不為空白就尋找到樹根的左孩子節點，然後尋找該節點是否有右孩子，如果有繼續尋找，直到找到屬於葉子節點的右孩子，此時，該節點的右子樹“指向”當前樹的右子樹，並將當前左子樹變為樹根的右孩子，將整棵樹左孩子置為空白。最後，根節點“指向”根節點的右孩子，繼續上述操作，直到整棵樹遍曆完即得到結果。

（3）希望本文對你有所協助。

演算法代碼實現如下：

/** * Definition for binary tree * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */public void flatten(TreeNode root) {while (root != null) {if (root.left != null) {TreeNode pre = root.left;while (pre.right != null)pre = pre.right;pre.right = root.right;root.right = root.left;root.left = null;}root = root.right;}}