代码随想录算法训练营第二十天| 654.最大二叉树、 617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树

Leetcode 54.最大二叉树

class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        TreeNode* node = new TreeNode(0);
        if(nums.size()==1){ //遇到叶子节点
            node->val = nums[0];
            return node;
        }
        int maxValue = 0; //数组中的最大值
        int index  = 0; //最大值对应的索引
        for(int i=0;i<nums.size();i++){
            if(nums[i]>maxValue){
                maxValue=nums[i];
                index = i;
            }
        }
        node->val = maxValue;
        if(index>0){ //最大值左边的区间有元素 构造左子树
            vector<int> newVec(nums.begin(),nums.begin()+index);
            node->left = constructMaximumBinaryTree(newVec);
        }
        if(index<nums.size()-1){ //最大值右边的区间有元素 构造右子树
            vector<int> newVec(nums.begin()+index+1,nums.end());
            node->right = constructMaximumBinaryTree(newVec);
        }
        return node;
    }
};

Leetcode 617.合并二叉树

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1==NULL)return root2; //如果节点1为空 合并之后是节点2
        if(root2==NULL)return root1; //如果节点2为空 合并之后是节点1
        //将节点1和节点2的值相加
        root1->val += root2->val; //中 
        root1->left = mergeTrees(root1->left, root2->left); //左
        root1->right = mergeTrees(root1->right, root2->right); //右
        return root1;
    }
};

Leetcode 700.二叉搜索树中的搜索

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root==NULL || root->val==val)return root;
        TreeNode* result = NULL;      
        if(root->val > val){ //节点值大于要搜索的值
            result = searchBST(root->left, val); //向左搜索
        }
         if(root->val < val){ //节点值小于要搜索的值
            result = searchBST(root->right, val); //向右搜索
        }
        return result;
    }
};

Leetcode 98.验证二叉搜索树

class Solution {
public:
    TreeNode* pre = NULL; //root前一个节点的指针 用于和root比较对应节点值的大小
    bool isValidBST(TreeNode* root) {   
        if(root==NULL)return true;
        bool left = isValidBST(root->left); //左 判断左子树是否是二叉搜索树
        if(pre != NULL && pre->val >= root->val)return false; //前一个节点比当前节点的值大
        pre = root; //记录前一个节点
        bool right = isValidBST(root->right); // 右 判断右子树是否是二叉搜索树
        return left && right; //节点的左右子树都是二叉搜索树才为true
    }
};