您的位置:首页 > 其它

lintcode:将二叉树拆成链表

2016-07-07 19:46 363 查看
题目

将一棵二叉树按照前序遍历拆解成为一个
假链表
。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的 next 指针。

注意事项

不要忘记将左儿子标记为 null,否则你可能会得到空间溢出或是时间溢出。

样例

1
\
1          2
/ \          \
2   5    =>    3
/ \   \          \
3   4   6          4
\
5
\
6


解题

修改前序遍历

/**
* Definition of TreeNode:
* public class TreeNode {
*     public int val;
*     public TreeNode left, right;
*     public TreeNode(int val) {
*         this.val = val;
*         this.left = this.right = null;
*     }
* }
*/
public class Solution {
/**
* @param root: a TreeNode, the root of the binary tree
* @return: nothing
* 前序遍历:根左右
* 修改根的right指向左部分
* 左部分的right指向右部分
*/
public void flatten(TreeNode root) {
// write your code here
// 空结点
if(root == null)
return;
// 左右结点都空
if(root.left==null && root.right==null)
return;
// 左结点空
if(root.left==null){
TreeNode right = root.right;
flatten(right);
return;
}
// 右节点空
if(root.right==null){
TreeNode left = root.left;
root.right = left;

flatten(left);
root.left = null;// left结点设置为空
return;
}
// 左右结点都不空
TreeNode left = root.left;
TreeNode right = root.right;
root.right = left;

flatten(left);

TreeNode p = left;
while(p.right!=null){ // 找到left部分的最右结点
p = p.right;
}
p.right = right;
flatten(right);

root.left = null; // left结点设置为空
return;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: