算法day16|654.最大二叉树、617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树

算法day16|654.最大二叉树、617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树

654.最大二叉树

这道题很简单,其实就是105、106的变式题。具体代码如下:

cpp 复制代码
class Solution {
public:
    TreeNode*traversal(vector<int>& nums)
    {
        if(nums.empty())
        return nullptr;
        int max=nums[0];
        int index=0;
        for(int i=1;i<nums.size();i++)
        {
            if(nums[i]>max)
            {
                max=nums[i];
                index=i;
            } 
        }
        TreeNode*root=new TreeNode(max);
        if(nums.size()==1)
        return root;
        vector<int> leftNums(nums.begin(),nums.begin()+index);
        vector<int> rightNums(nums.begin()+index+1,nums.end());
        root->left=traversal(leftNums);
        root->right=traversal(rightNums);
        return root;
    }
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        if(!nums.empty())
        return traversal(nums);
        else
        return nullptr;
    }
};

总体思路与105、106类似,甚至更简单。

617.合并二叉树

1.额外申请空间(失败)

不知道为什么运行不了...,代码如下:

cpp 复制代码
class Solution {
public:
    TreeNode* traversal(TreeNode* root1, TreeNode* root2)
    {
        if(root1==nullptr&&root2==nullptr)
        return nullptr;
        TreeNode*root=new TreeNode();
        if(root1!=nullptr&&root2!=nullptr)
        root->val=root1->val+root2->val;
        else if(root1==nullptr&&root2!=nullptr)
        root->val=root2->val;
        else if(root1!=nullptr&&root2==nullptr)
        root->val=root1->val;

        root->left=traversal(root1->left,root2->left);
        root->right=traversal(root1->right,root2->right);
        return root;

    }
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) 
    {
        return traversal(root1,root2);
    }
};

2.不额外申请空间

cpp 复制代码
class Solution {
public:
TreeNode* traversal(TreeNode* root1, TreeNode* root2)
    {
        if(!root1&&!root2)
        return nullptr;
        else if(!root1)
        return root2;
        else if(!root2)
        return root1;
        else{
            root1->val+=root2->val;
        }
        root1->left=traversal(root1->left,root2->left);
        root1->right=traversal(root1->right,root2->right);
        return root1;

    }
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        return traversal(root1,root2);
    }
};

直接在root1上进行操作,不用额外申请空间。

700.二叉搜索树中的搜索

cpp 复制代码
class Solution {
public:
    TreeNode* traversal(TreeNode* root, int val)
    {
        if(root==nullptr)
        return nullptr;
        if(root->val>val)
            return traversal(root->left,val);
        if(root->val<val)
            return traversal(root->right,val);
        else
            return root;
    }
    TreeNode* searchBST(TreeNode* root, int val) {
        return traversal(root,val);
    }
};

要注意一下BST 的特点:BST首先得是二叉平衡树,满足左<中<右。所以:

cpp 复制代码
if(root->val>val)
            return traversal(root->left,val);
        if(root->val<val)
            return traversal(root->right,val);

另外,如果递归有返回值的话,在单层递归里面肯定是需要设置变量来接收的,或者直接return 递归。

98.验证二叉搜索树

1.遍历后排序

cpp 复制代码
class Solution {
public:
    void traversal(TreeNode* root,vector<int> &vec)
    {
        if(root==nullptr)
        return ;
        traversal(root->left,vec);
        vec.push_back(root->val);
        traversal(root->right,vec);
    }

    bool isValidBST(TreeNode* root) {
        if(root==nullptr)
        return true;
        else
        {
            vector<int> vec;
            traversal(root,vec);
            for(int i=0;i<vec.size()-1;i++)
            {
                if(vec[i]>=vec[i+1])
                return false;
            }
            return true;
        }
    }
};

这题的易错点就是必须保证左子树上的所有元素都要小于根节点,右子树同理,而不是仅仅是单个左孩子结点或者右孩子结点。这样的思路用递归就很难实现了。

所以我们另辟蹊径,利用二叉搜索树的最重要的特征之一:中序序列单调递增 。我们只需要用数组收集中序序列,然后去判断它是否递增即可。

2.边遍历遍排序

对于递增的判断其实是可以在遍历过程中就实现的,代码如下:

cpp 复制代码
class Solution {
public:
    long long MaxValue = LONG_MIN;
    bool traversal(TreeNode* root)
    {
        if(root==nullptr)
        return true;
        bool left=traversal(root->left);
        if(root->val>MaxValue)
        MaxValue=root->val;
        else
        return false; 
        bool right=traversal(root->right);
        return left&&right;

    }
    bool isValidBST(TreeNode* root) {
        return traversal(root);
    }
};

为了判断中序序列是否递增,我们需要一个值来继承之前结点的大小,然后与当前结点比较即可。由于力扣中输入了Int的最小值,所以我们采用long long型的最小值来接受第一个元素,十分巧妙:

cpp 复制代码
		long long MaxValue = LONG_MIN;
        if(root->val>MaxValue)
                MaxValue=root->val;
                else
                return false;

这样就可以保证第一个数的顺利进行。因为不想for循环,递归的时候是很难定位到第几个元素的,所以想把第哪个元素赋值为几,这是做不到的。只有在深刻理解逻辑之后做一些巧思。

3.指针记录法

cpp 复制代码
class Solution {
public:
    TreeNode*pre=nullptr;
    bool traversal(TreeNode* root)
    {
        if(root==nullptr)
        return true;
        bool left=traversal(root->left);
        if(pre!=nullptr&&pre->val>=root->val)
        return false;
        else
        pre=root;
        bool right=traversal(root->right);
        return left&&right;

    }
    bool isValidBST(TreeNode* root) {
        return traversal(root);
    }
};

我们改用指针来记录,实现一个更巧妙的逻辑:用指针记录前面的值,只有当指针的值大于当前的值时,return false。问题在于,指针该如何记录呢?逻辑很巧妙,因为记录第一个值是非常关键的。初始时,我们设pre为null,第一次就是因为pre为null,成功赋值;其他时候是因为满足排序,所以成功赋值。它们的逻辑是有区别的。

相关推荐
To_OC2 小时前
LC 49 字母异位词分组:想到哈希表很简单,选对 key 才是精髓
javascript·算法·leetcode
用户938515635077 小时前
从 O(n²) 到 O(nlogn):一文读懂快速排序的“快”与“妙”
javascript·算法
To_OC8 小时前
手写快排次次翻车?别死背快排模板了,这才是面试官想听的底层逻辑
javascript·算法·排序算法
饼干哥哥8 小时前
Reddit VOC调研太慢?搭一个AI专家团队半小时洞察任何品类|以猫用饮水机为例
人工智能·算法·ai编程
地平线开发者10 小时前
Transformer模型部署之性能优化指南
算法
地平线开发者10 小时前
人在途中:从“编译失败”到“模型可落地”——CUDA 自定义算子
算法·自动驾驶
半个落月13 小时前
从递归到快速排序:用 JavaScript 把分治思想讲明白
javascript·算法·面试
小月土星14 小时前
JavaScript 快速排序:从 pivot、双指针到分治思想
javascript·算法·面试
小月土星14 小时前
JavaScript 递归入门:从 1 到 n 求和,再到数组扁平化
javascript·算法·面试