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;
}
}
}
}