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

最新推荐文章于 2024-10-14 18:52:56 发布

chstor

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量101

点赞数

文章标签： leetcode 深度优先算法

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

版权

本文介绍了如何使用递归和迭代方法判断一棵二叉树是否对称，以及如何根据前序和中序遍历序列来构造二叉树。主要涉及二叉树的特性和遍历策略。

摘要由CSDN通过智能技术生成

101. 对称二叉树

递归写法

bool isSymmetric(TreeNode* root) {
        if(!root) return true;
        return dfs(root->left,root->right);
    }
bool dfs(TreeNode* p,TreeNode* q){
        if(!p || !q) return !p && !q;
        return p->val==q->val && dfs(p->left,q->right) && dfs(p->right,q->left);
}

迭代写法

    bool isSymmetric(TreeNode* root) {
        if(!root) return true;
        stack<TreeNode*> p,q;
        auto l = root->left,r = root->right;
        while(l || r || p.size() || q.size()){
            while(l && r){
                p.push(l);
                q.push(r);
                l=l->left;
                r=r->right;
            }
            if(l || r) return false;
            l = p.top(),p.pop();
            r = q.top(),q.pop();
            if(l->val != r->val) return false;
            l = l->right,r = r->left;
        }
        return true;
    }

105. 从前序与中序遍历序列构造二叉树

在这里插入图片描述

    unordered_map<int,int> pos;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        for(int i = 0; i < n; i ++) pos[inorder[i]] = i;
        return dfs(preorder,inorder,0,n-1,0,n-1);
    }
    TreeNode* dfs(vector<int>& preorder,vector<int>& inorder,int pl,int pr,int il,int ir){
        if(pl > pr) return NULL;
        int val = preorder[pl];
        int k = pos[val];
        int len = k-il;
        auto root = new TreeNode(val);
        root->left = dfs(preorder,inorder,pl+1,pl+len,il,k-1);
        root->right = dfs(preorder,inorder,pl+len+1,pr,k+1,ir);
        return root;
    }