码迷,mamicode.com
首页 > 其他好文 > 详细

Flatten Binary Tree to Linked List

时间:2015-01-22 19:38:34      阅读:219      评论:0      收藏:0      [点我收藏+]

标签:

随笔一记,留做重温!

 

Flatten Binary Tree to Linked List

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 static void flatten(TreeNode root) {
        if(root==null){
            return ;
        }
        TreeNode temp=root;
        Queue<TreeNode>queue=new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            TreeNode topNode=queue.poll();
            if(topNode.left!=null){
                queue.add(topNode.left);
            }
            if(topNode.right!=null){
                queue.add(topNode.right);
            }
            topNode.left=null;
            topNode.right=null;
            
            temp.right=topNode;
            temp.left=null;
            temp=temp.right;
            
        }
    }

 

 

 结果空间复杂度太高。然后我们参照网上给出的一种思路。

将树拆开。root,root.left(左子树),root.right(右子树)3部分,然后将右子树接在左子树的最右结点(右指针)上。同时,使得root的right指向root.left

root.left=null

root=root.right(下一个过程,循环)

public static void flatten2(TreeNode root) {
        if(root==null){
            return ;
        }
        while(root!=null){
            TreeNode leftTreeNode=root.left;
            TreeNode rightTreeNode=root.right;
            if(leftTreeNode!=null){
                TreeNode rightmosTreeNode=leftTreeNode;
                while(rightmosTreeNode.right!=null){
                    rightmosTreeNode=rightmosTreeNode.right;
                }
                rightmosTreeNode.right=rightTreeNode;
                root.right=leftTreeNode;
            }
            root.left=null;
            root=root.right;
        }
        
    }

 

Flatten Binary Tree to Linked List

标签:

原文地址:http://www.cnblogs.com/hitkb/p/4242204.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!