ACwing 刷题 2022/3/10

脑瓜嗡嗡0608

已于 2022-03-10 10:07:14 修改

阅读量921

点赞数

分类专栏： ACwing 文章标签： c++ 算法

于 2022-03-10 10:06:40 首次发布

本文链接：https://blog.csdn.net/weixin_53356651/article/details/123393349

版权

ACwing 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

只记录做了的题目和感觉有收获的题目
今日完成：
在这里插入图片描述

18.重建二叉树

好久没做这种基础题了，感觉都不会了……果然基础很重要。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<int,int> pos;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int len=preorder.size();
        for(int i=0;i<len;i++)
            pos[inorder[i]]=i;//方便根据先序遍历中的根节点位置在中序遍历中找到相应位置
        return dfs(preorder,inorder,0,len-1,0,len-1);
    }
    TreeNode* dfs(vector<int>& preorder, vector<int>& inorder,int pl,int pr,int il,int ir){
        if(pl>pr||il>ir) return NULL;
        if(pl<0||il<0) return NULL;
        int k=pos[preorder[pl]]-il;//计算出左子树的节点个数
        TreeNode* root=new TreeNode(preorder[pl]);//以当前根节点为对象建立一棵树
        root->left=dfs(preorder,inorder,pl+1,pl+k,il,il+k-1);
        root->right=dfs(preorder,inorder,pl+k+1,pr,il+k+1,ir);
        return root;
    }
};

模板题目，记忆方法。如果看不懂就建树模拟

19.二叉树的下一个节点

主要思路：中序遍历是左根右，那么如果待求节点有右孩子，则待求节点中序遍历的下一个节点应当是其右孩子的最左节点。如果没有，那么待求节点中序遍历的下一个节点应当是向上的某个p节点，该节点是第一个是其father左孩子的节点。不懂就画图尝试。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode *father;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL), father(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* p) {
        //中：左、根、右
        if(p->right){
            p=p->right;
            while(p->left)
                p=p->left;
            return p;
        }
        while(p->father&&p==p->father->right) p=p->father;
        return p->father;
    }
};