剑指 Offer II 052. 展平二叉搜索树

最新推荐文章于 2022-12-31 08:58:30 发布

还记得樱花正开~

最新推荐文章于 2022-12-31 08:58:30 发布

阅读量264

点赞数

分类专栏： leetcode 文章标签： leetcode 深度优先

本文链接：https://blog.csdn.net/ZHUYAN1209/article/details/122929145

版权

leetcode 专栏收录该内容

223 篇文章 2 订阅

订阅专栏

题目

力扣

思路一递归

中序遍历把节点存到vector里面，再逐个取出来，连接成新的二叉树。

代码一

class Solution {
public:
    vector<TreeNode*> v;
    TreeNode* increasingBST(TreeNode* root) {
        dfs(root);
        TreeNode* dummy = new TreeNode(-1);
        TreeNode* cur = dummy;
        for(TreeNode* node:v){
            cur->right = node;
            node->left = nullptr;
            cur = node;
        }
        return dummy->right;
    }
    void dfs(TreeNode* root){
        if(root == nullptr) return;
        dfs(root->left);
        v.push_back(root);
        dfs(root->right);
    }
};

思路二迭代

在栈中逐个存入左孩子，存完之后再pop出第一个，放到列表中，再遍历右子树，实现迭代式中序遍历。

代码二

class Solution {
public:
    TreeNode* increasingBST(TreeNode* root) {
        vector<TreeNode*> v;
        stack<TreeNode*> s;
        while(root != nullptr || !s.empty()){
            while(root){
                s.push(root);
                root = root->left;
            }
            root = s.top();
            s.pop();
            v.push_back(root);
            root = root->right;
        }
        TreeNode* dummy = new TreeNode(-1);
        TreeNode* cur = dummy;
        for(TreeNode* node:v){
            cur->right = node;
            node->left = nullptr;
            cur = node;
        }
        return dummy->right;
    }
};

思路三 Morris中序遍历

定义前驱节点tail，表示是上一个cur。定义dummy节点，dummy->right就表示头节点。

1.如果当前节点cur的左节点为空，说明左子树为空或者遍历完了，就让tail的右指针指向cur，并且更新tail，继续遍历cur的右子树。

2.如果当前节点cur的左节点不为空，就让pre找到cur左子树的最右节点。

2.1如果最右节点的右指针为空，让最右节点指向cur，继续遍历cur的左子树。

2.2如果最右节点的右指针指向cur，说明已经遍历过一遍了，把cur的左指针置为空，继续遍历cur的右子树，并且让tail右指针指向cur，更新tail。

代码三

class Solution {
public:
    TreeNode* increasingBST(TreeNode* root) {
        TreeNode* pre = nullptr, *cur = root;
        TreeNode* dummy = new TreeNode(0), *tail = dummy;
        while(cur){
            pre = cur->left;
            if(pre){
                while(pre->right != nullptr && pre->right != cur)
                    pre = pre->right;
                if(pre->right == nullptr){
                    pre->right = cur;
                    cur = cur->left;
                    continue;
                }else{
                    // pre->right = nullptr;
                    cur->left = nullptr;
                }
            }
            tail->right = cur;
            tail = cur;
            cur = cur->right;
        }
        return dummy->right;
    }
};