/*114. 二叉树展开为链表
给定一个二叉树,原地将它展开为链表。
例如,给定二叉树
1
/ \
2 5
\
4
将其展开为:
1
\
2
\
4
\
5
* */
// 他法:在还没操作节点右子树前,不能破坏该节点的右子树指向。所以采用后序遍历。
public void flatten(TreeNode root) {
if (root == null)
return;
flatten(root.left);
flatten(root.right);
TreeNode tmp = root.right;
root.right = root.left;
root.left = null;
while (root.right != null)
root = root.right;
root.right = tmp;
}
// 自己的方法,已通过(先序遍历递归法)
public void flatten2(TreeNode root) {
if (root == null)
return;
flattenCore(root);
}
public TreeNode flattenCore(TreeNode root) {
TreeNode ret = root;
if (root == null)
return null;
if (root.left != null && root.right != null) {
// 先找到最底层节点
TreeNode temp = root.right;
root.right = flattenCore(root.left);
root.left = null;
while (root.right != null) {
root = root.right;
root.left = null;
}
root.right = flattenCore(temp);
root.left = null;
return ret;
}
if (root.left != null) {
root.right = flattenCore(root.left);
root.left = null;
}
if (root.right != null) {
root.right = flattenCore(root.right);
}
return root;
}
LeetCode114—— 二叉树展开为链表
最新推荐文章于 2024-05-01 21:09:02 发布