[Leetcode] Flatten Binary Tree to Linked List

最新推荐文章于 2022-09-06 16:04:14 发布

原创最新推荐文章于 2022-09-06 16:04:14 发布 · 275 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

89 篇文章

订阅专栏

本文介绍了一种将二叉树结构通过深度优先搜索（DFS）算法展平为链表的方法。具体实现中，首先递归地展平左子树和右子树，然后找到左子树的最右节点，并将其右指针指向原来的右子树；最后将根节点的左子树置空，而右子树则更新为原本的左子树。

题目：

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

For example,
Given

The flattened tree should look like:

思路：DFS，返回后左右子树已经flatten完毕，这个时候只有要找到左子树的的叶子节点，把右子树接过去，然后整体作为root的右子树即可。

class Solution {
public:
    void flatten(TreeNode *root) {
        if (root == nullptr) return;
        flatten(root->left);
        flatten(root->right);
        if (root->left == nullptr) return;
        TreeNode* tail = root->left;
        while (tail->right != nullptr) tail = tail->right;
        tail->right = root->right;
        root->right = root->left;
        root->left = nullptr;
    }
};

总结：复杂度为O(2^n).