首页 > 代码库 > Flatten Binary Tree to Linked List
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 6The flattened tree should look like:
1 2 3 4 5 6
click to show hints.
Hints:
If you notice carefully in the flattened tree, each node‘s right child points to the next node of a pre-order traversal.
空间消耗比较大,非in-place的解法。
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { public void flatten(TreeNode root) { if(root==null) return; ArrayList<TreeNode> a=new ArrayList<TreeNode>(); prevOrder(root,a); for(int i=0;i<a.size()-1;i++){ a.get(i).left=null; a.get(i).right=a.get(i+1); } } public void prevOrder (TreeNode root,List<TreeNode> a){ if(root==null) return; a.add(root); prevOrder(root.left,a); prevOrder(root.right,a); } }改进的解法
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { public void flatten(TreeNode root) { if(root==null) return; while(root!=null){ if(root.left!=null){ TreeNode node=root.left; while(node.right!=null) node=node.right; node.right=root.right; root.right=root.left; root.left=null; } root=root.right; } } }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。