LeetCode 114. 二叉树展开为链表
题目
给定一个二叉树,原地将他展开为一个链表(单向,右指针)。
如图:
方法1:前序遍历递归
class Solution {
TreeNode pre; //用来保存当前访问的节点的前驱节点
public void flatten(TreeNode root) {
if(root == null){
return;
}
Preorder(root);
}
public TreeNode Preorder(TreeNode root){
TreeNode right_node = root.right;
pre = root; // 更新前驱节点
if(root.left != null){
root.right = Preorder(root.left);
root.left = null;
}
if(right_node != null){
pre.right = Preorder(right_node);
}
return root;
}
}
方法2:前序遍历循环
注意到前序遍历访问各节点的顺序是根节点、左子树、右子树。如果一个节点的左子节点为空,则该节点不需要进行展开操作。如果一个节点的左子节点不为空,则该节点的左子树中的最后一个节点被访问之后,该节点的右子节点被访问。该节点的左子树中最后一个被访问的节点是左子树中的最右边的节点,因此将当前访问节点的右子树移动到左子树的最右,左子树在遍历完原有节点后,必遍历原访问节点的右子树。因此,问题转化成寻找右子树的插入位置。
具体做法是,对于当前节点,如果其左子节点不为空,则在其左子树中找到最右边的节点,作为前驱节点,将当前节点的右子节点赋给前驱节点的右子节点,然后将当前节点的左子节点赋给当前节点的右子节点,并将当前节点的左子节点设为空。对当前节点处理结束后,继续处理链表中的下一个节点,直到所有节点都处理结束。
该方法可以将空间复杂度从O(n)优化到O(1)。
class Solution {
public void flatten(TreeNode root) {
TreeNode curr = root;
while (curr != null) {
if (curr.left != null) { //左子树不为空时
TreeNode next = curr.left;
TreeNode predecessor = next;
while (predecessor.right != null) { //找到左子树中最右的位置
predecessor = predecessor.right;
}
predecessor.right = curr.right; //将右子树嫁接
curr.left = null;
curr.right = next;
}
curr = curr.right;
}
}
}