LeetCode刷题之HOT100之二叉树展开为链表

2024/6/17 早上好呀，今天周一，又刷新了。这周是第十七周，还有一周就结课了，最近组里没啥事，期末课设做一做，刷刷题，随便乱学学。ok，开始新的一周的新的一题吧，从今天起，做个幸福的人吧。

1、题目描述

2、逻辑分析

题目要求就是将二叉树转换为链表形式。我的想法就是利用先序遍历思想进行求解。大致思路：

1. 首先，先序遍历整个二叉树，将每个节点储存在一个列表里，列表里面的数就是有序的；
2. 然后，遍历列表，将元素逐一放在右子树上即可

3、代码演示

public void flatten(TreeNode root) {
        List<TreeNode> list = new ArrayList<>();
        // 使用前序遍历将二叉树的节点依次添加到列表中 
        preOrder(root, list);
        // 获取列表中节点的数量  
        int n = list.size(); 
        // 遍历列表，从第二个节点开始（索引为1），将前一个节点的左子节点设为null，右子节点设为当前节点  
        // 从而将二叉树拉平为一个只有右子节点的链表   
        for(int i  = 1; i < n; i++){
         TreeNode pre = list.get(i - 1), cur = list.get(i);
         // 将前一个节点的左子节点设为null
         pre.left = null;
         // 将前一个节点的右子节点设为当前节点
         pre.right = cur;
        }
    }
    // 前序遍历二叉树的方法，将遍历到的节点添加到列表中 
    public void preOrder(TreeNode root, List<TreeNode> list){  
        // 如果节点不为空，则将其添加到列表中  
            if(root != null){
             list.add(root);
             // 递归遍历左子树  
            preOrder(root.left, list);
            // 递归遍历右子树
            preOrder(root.right, list);
        }
    }

以上代码结合了先序遍历和递归思想，简洁易理解。还有一种迭代方法就不过赘述，使用了栈来存储迭代。

4、复杂度分析

• 时间复杂度：O(n)。先序遍历需要的时间为O(n)，遍历之后在变成链表的时间复杂度也是O(n)
• 空间复杂度：O(n)。先序遍历需要的空间也是O(n)，因为递归的底层也使用到了栈。

有没有一种方法可以将空间复杂度降低为常数级别呢。是有的
解题思路：
前序遍历访问各节点的顺序是根节点、左子树、右子树。
对于当前节点，如果其左子节点不为空，则在其左子树中找到最右边的节点，作为前驱节点。
将当前节点的右子节点赋给前驱节点的右子节点，然后将当前节点的左子节点赋给当前节点的右子节点，并将当前节点的左子节点设为空。
对当前节点处理结束后，继续处理链表中的下一个节点，直到所有节点都处理结束。

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;  
    }  
}

ok，水完了，拜拜啦！