Leetcode刷题笔记(c++)_剑指 Offer 36. 二叉搜索树与双向链表

最新推荐文章于 2024-11-13 11:37:20 发布

大风吹~~~~~

最新推荐文章于 2024-11-13 11:37:20 发布

阅读量345

点赞数

本文链接：https://blog.csdn.net/LW_12345/article/details/124571824

版权

中序遍历

用只记录前一个和第一个节点记录

class Solution {
public:
    Node* pre;Node* head;
    void dfs(Node* root){
        if(!root)return;
        dfs(root->left);
        if(pre){
            pre->right=root;
            root->left=pre;
        }else head=root;
        pre=root;
        dfs(root->right);
    }
    Node* treeToDoublyList(Node* root) {
        dfs(root);
        if(!root)return root;
        pre->right=head;
        head->left=pre;
        return head;       
    }
};

在这里插入图片描述

中序遍历

用vector记录

class Solution {
public:
    vector<Node*>temp;
    void dfs(Node* root){
        if(!root)return;
        dfs(root->left);
        temp.push_back(root);
        dfs(root->right);
    }
    Node* treeToDoublyList(Node* root) {
        dfs(root);
        if(!root)return root;
        int n=temp.size();
        // for(Node* ab:temp)cout<<ab->val<<endl;
        for(int i=1;i<n;i++){
            temp[i-1]->right=temp[i];
            temp[i]->left=temp[i-1];
        }
        temp[n-1]->right=temp[0];
        temp[0]->left=temp[n-1];
        return temp[0];       
    }
};