代码随想录跟练日记|Day17 二叉树part05

算法小白训练营日记，笔记为自用，若有错误感谢指出

--今日任务--：654.最大二叉树 ，617.合并二叉树 ，700.二叉搜索树中的搜索 ，98.验证二叉搜索树

654.最大二叉树

题目链接：654. 最大二叉树 - 力扣（LeetCode）

解题思路：遍历数组寻找最大值，将值赋给新节点，并记录其下标，然后将剩余数组按最大值下标分割为左右两个新数组，继续寻找最大值记录下标，其值为根节点的左右子节点，再继续分割数组。依次递归，构造二叉树，最后返回根节点node。

具体代码如下：

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 maxValueIndex=0;
        for(int i=0;i<nums.size();i++){
            if(nums[i]>maxValue){
                maxValue=nums[i];
                maxValueIndex=i;
            }
        }
        node->val=maxValue;
        if(maxValueIndex>0){
            vector<int> newVec(nums.begin(),nums.begin()+maxValueIndex);
            node->left=constructMaximumBinaryTree(newVec);
        }
        if(maxValueIndex<(nums.size()-1)){
            vector<int> newVec(nums.begin()+maxValueIndex+1,nums.end());
            node->right=constructMaximumBinaryTree(newVec);
        }
        return node;
    }
};

617.合并二叉树

题目链接：617. 合并二叉树 - 力扣（LeetCode）

解题思路：递归法合并两个二叉树，运用前序，先处理中间节点，若r1为空则返回r2，r2为空则返回r1，中间节点值为r1r2值之和，再依次递归遍历中间节点的左右子树，最后返回root。

具体代码如下：

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1==nullptr) return root2;
        if(root2==nullptr) return root1;
        TreeNode* root=new TreeNode(0);
        root->val=root1->val+root2->val;
        root->left=mergeTrees(root1->left,root2->left);
        root->right=mergeTrees(root1->right,root2->right);
        return root;
    }
};

700.二叉搜索树中的搜索

题目链接：700. 二叉搜索树中的搜索 - 力扣（LeetCode）

解题思路：比较根节点与val值大小，val值小查找根节点的左节点，val值大查找右节点，依次查找得到结果。

具体代码如下：

递归法：

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root==NULL||val==root->val) return root;
        TreeNode* node=NULL;
        if(val<root->val) node=searchBST(root->left,val);
        if(val>root->val) node=searchBST(root->right,val);
        return node;
    }
};

迭代法：

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        while(root!=nullptr){
            if(val<root->val) root=root->left;
            else if(val>root->val) root=root->right;
            else return root;
        }
        return NULL;
    }
};

98.验证二叉搜索树

题目链接：98. 验证二叉搜索树 - 力扣（LeetCode）

解题思路：递归法：

方法一：中序遍历二叉树，将节点值存入数组，比较数组是否按序排列。代码如下：

class Solution {
private:
    vector<int> vec;
    void traversal(TreeNode* root){
        if(root==NULL) return;
        traversal(root->left);
        vec.push_back(root->val);
        traversal(root->right);
    }

public:
    bool isValidBST(TreeNode* root) {
        vec.clear();
        traversal(root);
        for(int i=1;i<vec.size();i++){
            if(vec[i]<=vec[i-1]) return false;
        }
        return true;
    }
};

方法二：依次比较每个节点与它前一个节点的大小，如果前节点的值大于等于当前节点，则说明该二叉树不是二叉搜索树，返回false。代码如下：

class Solution {
public:
    TreeNode* pre=NULL;
    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;
    }
};