首页 > 其他 > 详细

leetcode Flatten Binary Tree to Linked List

时间:2017-08-20 09:27:41      阅读:224      评论:0      收藏:0      [点我收藏+]

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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!