[LeetCode]Flatten Binary Tree to Linked List

CiaoLiang

于 2015-04-17 20:54:26 发布

阅读量364

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/CiaoLiang/article/details/45101981

LeetCode 专栏收录该内容

149 篇文章

订阅专栏

本文介绍了一种将二叉树结构通过特定算法展平为链表的方法。该方法利用递归特性，首先记录当前节点的左右子树，然后调整节点连接，确保每个节点的左子树为空且右子树按前序遍历顺序排列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

The flattened tree should look like:

分析：最后生成的类似链表的树状结构就是树的前序遍历。关于树结构的编程用递归都很好写。我们采用类似前序遍历的方法，先把树右节点改为左侧的前序遍历，然后再找到左侧的最后一个节点，右节点链接到右侧的前序遍历。

特别注意边界条件，特别是左树为NULL的时候。代码如下：

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode *root) {
        if(root==NULL)
            return;
        TreeNode* KeepRight = root->right;//记录节点
        root->right = root->left;
        TreeNode* KeepLeft = root->left;
        TreeNode* pre = root;
        root->left = NULL; //左侧断链
        flatten(KeepLeft);
        while(pre->right!=NULL)//找到左侧最后一个节点，注意左侧为NULL的情况
           pre = pre->right;
        pre->right = KeepRight;
        flatten(KeepRight);
    }
};