[LeetCode] Flatten Binary Tree to Linked List
阿新 • • 發佈：2017-05-26
mil neu pan gin 一次 ng- mon example style
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. 反復1。2
思路不難，代碼上有點混亂。細節實現是：把root的左子樹弄成flatten的 flatten(root.left)，把右子樹保存起來 temp=root.right，讓root.right=root.left, 然後斷掉左邊,root.left=null.
然後一直走到右邊的最末端，while(root.right!=null) root=root.right 然後root.right=temp，繼續處理。
一次通過。代碼例如以下：
public void flatten(TreeNode root) {
if(root==null) return;
TreeNode p=root;
while(p!=null){
if(p.left==null){
p=p.right;
} else {
if(p.right==null) {
TreeNode temp=p.left;
p.left=null;
p.right=temp;
p=p.right;
} else {
TreeNode tempRight=p.right;
TreeNode tempLeft=p.left;
p.right=null;
flatten(p.left);
p.left=null;
p.right=tempLeft;
while(p.right!=null){
p=p.right;
}
p.right=tempRight;
}
}
}
}[LeetCode] Flatten Binary Tree to Linked List