【LeetCode】114. 二叉树展开为链表

最新推荐文章于 2025-09-27 23:15:14 发布

原创最新推荐文章于 2025-09-27 23:15:14 发布 · 216 阅读

0 ·

CC 4.0 BY-SA版权

OJ刷题专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种将二叉树原地展平为单链表的方法，通过递归处理左、右子树并将左子树的最后一个节点指向右子树，实现树结构的链表化。代码示例使用C++实现。

文章目录

1.题目

给定一个二叉树，原地将它展开为一个单链表。

例如，给定二叉树

将其展开为：

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list

2.思路

题目本质就是把树的左子树全部并到右子树中，虽然没说顺序，但按示例，就是左-中-右了。也就是说，要把root的右子树，放到左子树的最后一个结点的右子树中。至于左右子树，各自的处理，就是标准的递归了。

所以递归函数的步骤可以理解如下：

当root为空时，不处理；
递归处理root的左子树；递归处理root的右子树；(当然左右子树也可以后面再处理)
当root没有左子树时，无需处理；
当root有左子树时，要找到最后一个结点last，从root.left开始往下找（因为是已经处理完的，所以root.left只有右子树），循环找right直到最后一个结点即可；
关键：
- 将root的右子树移到last的右指针，last.right = root.right;
- root的左子树移到root的右指针:root.right = root.left;
- 清空root的左指针：root.left = null；

作者：gousiqi
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/solution/na-ge-wai-mai-de-gong-fu-tu-ran-xiang-tong-liao-by/
来源：力扣（LeetCode）

所以，面对一个问题的时候，理解问题本质和拆分子问题是最重要的。思路明晰了，代码自然就出来了。

3.代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        if(root==nullptr)return;
        flatten(root->left);
        flatten(root->right);
        if(root->left!=nullptr){
            //找到左子树的最后一个节点
            TreeNode* last=root->left;
            while(last->right!=nullptr)last=last->right;
            //把右子树添加为左子树最后一个节点右子树
            last->right=root->right;
            //把左子树添加为根的右子树
            root->right=root->left;
            //把根节点的左子树置为空
            root->left=nullptr;
        }
    }
};