leetcode114. Flatten Binary Tree to Linked List-CSDN博客

本文链接：https://blog.csdn.net/weixin_43216252/article/details/121081986

思路：题目大意就是求树的前序遍历。这点观察是可以发现的。所以只要把前序遍历的结果放入数组，然后把他们串起来就可以了。

方法一前序遍历

有两种，递归和非递归。看代码吧：

/**
 * 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:
    vector<TreeNode*>nodes;
    void preorderByRecursion(TreeNode* root) {//RLF root left right
        if (!root) return;
        nodes.push_back(root);
        preorderByRecursion(root->left);
        preorderByRecursion(root->right);
    }
    void preorderByIteration(TreeNode* root) {
        if (!root) return;
        stack<TreeNode*>s;
        s.push(root);
        while (!s.empty()) {
            TreeNode* node = s.top();
            s.pop();
            nodes.push_back(node);
            if (node->right) s.push(node->right);//根据栈先进后出的性质，把右结点先放进去就是先访问左结点。
            if (node->left) s.push(node->left);
        }
    }
    void flatten(TreeNode* root) {
        //preorderByRecursion(root);
        preorderByIteration(root);
        TreeNode* prev = nullptr;
        for (auto& node : nodes) {//把他们串起来
            if (prev) {
                prev->right = node;
                prev->left = nullptr;
            }
            prev = node;
        }
    }
};

方法二找前驱结点

我们只要把每个结点的前驱结点找到，然后把后面的放在前驱结点的右边，其左边置为空指针。在更新当前前驱结点就可以了。看代码吧：

/**
 * 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:
    TreeNode* pre = nullptr;
    void preorder(TreeNode* root) {
        if (!root) return;
        TreeNode* l = root->left;//先记录其左右结点
        TreeNode* r = root->right;
        if (pre) {//如果不是第一次执行，更新该前驱结点的右结点为当前结点，左结点为空。
            pre->right = root;
            pre->left = nullptr;
        }
        pre = root;//更新前驱结点。
        preorder(l);//因为是前序遍历
        preorder(r);
    }
    void flatten(TreeNode* root) {
        preorder(root);
    }
};