leetcode114.二叉树展开为链表(java)

最新推荐文章于 2024-08-23 14:25:42 发布

原创最新推荐文章于 2024-08-23 14:25:42 发布 · 225 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

87 篇文章

订阅专栏

本文深入探讨了一种将二叉树转换为链表的算法，通过修改树节点的左右指针，使得所有节点的左子节点为空，而右子节点形成一条从根节点到叶子节点的链表。这一过程不仅保留了原始节点的顺序，还有效地减少了空间复杂度。

题目
在这里插入图片描述

官方题解好有意思

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public void flatten(TreeNode root) {
        while(root != null){
            //找左子树的最右节点
            //如果不存在左子树
            if(root.left == null){
                root = root.right;
            }else{
                //找到左子树的最右节点
                TreeNode pre = root.left;
                while(pre.right != null){
                    pre = pre.right;
                }
                //该节点的右节点接右子树
                pre.right = root.right;
                // 根节点的右子树更新为左子树
                root.right = root.left;
                // 根节点的左子树记得置空！！！
                root.left = null;
                //根节点指向下一个节点
                root = root.right;
            }
        }

    }
}