[LeetCode] Flatten Binary Tree to Linked List

标签：面试题   leetcode   算法   tree   java   

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

For example,
Given

         1
        /        2   5
      / \        3   4   6

   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

标签：面试题   leetcode   算法   tree   java   

原文地址：http://blog.csdn.net/jhbxlx/article/details/39850823