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

leetcode Flatten Binary Tree to Linked List

时间:2017-08-20 10:09:07      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:new   void   binary   initial   att   nbsp   null   ack   tac   

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


解法:
iterative:
public void flatten(TreeNode root) {
        if (root==null) {
        return;
    }
   Stack<TreeNode> stack=new Stack<TreeNode>();
    stack.push(root);
   while(!stack.isEmpty()){
       TreeNode current=stack.pop();
       if (current.right!=null)  
           stack.push(current.right);
      if (current.left!=null)  
          stack.push(current.left);
      if (!stack.isEmpty()) 
          current.right = stack.peek();
      current.left=null;
      
   }
    }

recursive:

 public void flatten(TreeNode root) {
        if (root==null) {
        return;
    }
   TreeNode left=root.left;
   TreeNode right=root.right;
   flatten(left);
   flatten(right);
   root.left=null;
   root.right=left;
   TreeNode tmp=root;
   while(tmp.right!=null)
   {
       tmp=tmp.right;
   }
   tmp.right=right;
    }

 

leetcode Flatten Binary Tree to Linked List

标签:new   void   binary   initial   att   nbsp   null   ack   tac   

原文地址:http://www.cnblogs.com/lilyfindjobs/p/7398693.html

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