LeetCode暑期刷题打卡——树专题day5

最新推荐文章于 2024-09-29 16:57:04 发布

chstor

最新推荐文章于 2024-09-29 16:57:04 发布

阅读量96

点赞数

分类专栏： leetcode题目文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_45821690/article/details/130297721

版权

leetcode题目专栏收录该内容

13 篇文章 0 订阅

订阅专栏

173. 二叉搜索树迭代器

在这里插入图片描述

用栈来实现迭代

stack<TreeNode*> stk;
    BSTIterator(TreeNode* root) {
        while(root){
            stk.push(root);
            root = root->left;
        }
    }
    
    int next() {
        auto p = stk.top();
        stk.pop();
        int val = p->val;
        p = p->right;
        while(p){
            stk.push(p);
            p=p->left;
        }
        return val;
    }
    
    bool hasNext() {
        return !stk.empty();
    }

297. 二叉树的序列化与反序列化

在这里插入图片描述


    // Encodes a tree to a single string.
    string serialize(TreeNode* root) {
        string res;
        dfs1(root,res);
        return res;
    }

    void dfs1(TreeNode* root,string &res){
        if(!root){
            res += "#,";
            return ;
        }
        res += to_string(root->val) + ",";
        dfs1(root->left,res);
        dfs1(root->right,res);
    }

    // Decodes your encoded data to tree.
    TreeNode* deserialize(string data) {
        int u = 0;
        return dfs2(data,u);
    }

    TreeNode* dfs2(string &data,int &u){
        if(data[u] == '#'){
            u += 2;
            return NULL;
        }
        int t = 0;
        bool is_minus = false;
        if(data[u] == '-'){
            is_minus = true;
            u ++;
        }
        while(data[u] != ','){
            t = t * 10 + data[u] - '0';
            u ++;
        }
        u ++;
        if(is_minus) t = -t;
        auto root = new TreeNode(t);
        root->left = dfs2(data,u);
        root->right = dfs2(data,u);
        return root;
    }