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

题目链接：654.最大二叉树

这里有三步工作

1. 先要找到数组中最大的值和对应的下标， 最大的值构造根节点，下标用来下一步分割数组。

代码如下：

int maxValue =0;
int maxValueIndex =0;
for(int i =0; i < nums.size(); i++)
{
    if(nums[i]> maxValue)
    {
        maxValue = nums[i];
        maxValueIndex = i;
    }
}
TreeNode* node =newTreeNode(0);
node->val = maxValue;

2. 最大值所在的下标左区间 构造左子树

这里要判断maxValueIndex > 0，因为要保证左区间至少有一个数值。

代码如下：

if(maxValueIndex >0)
{
    vector<int>newVec(nums.begin(), nums.begin()+ maxValueIndex);
    node->left =constructMaximumBinaryTree(newVec);
}

3. 最大值所在的下标右区间 构造右子树

判断maxValueIndex < (nums.size() - 1)，确保右区间至少有一个数值。

代码如下：

if(maxValueIndex <(nums.size()-1))
{
    vector<int>newVec(nums.begin()+ maxValueIndex +1, nums.end());
    node->right =constructMaximumBinaryTree(newVec);
}

完整代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
    TreeNode* node=new TreeNode(0);
    if(nums.size()==1)
    {
        node->val=nums[0];
        return node;
    }
    int max=0;
    int maxIndex=0;
    for(int i=0;i<nums.size();i++)
    {
        if(nums[i]>max)
        {
            max=nums[i];
            maxIndex=i;
        }
    }
    node->val=max;
    if(maxIndex>0)
    {
        vector<int> newVec(nums.begin(),nums.begin()+maxIndex);
        node->left=constructMaximumBinaryTree(newVec);
    }
        if(maxIndex<nums.size()-1)
    {
        vector<int> newVec(nums.begin()+maxIndex+1,nums.end());
        node->right=constructMaximumBinaryTree(newVec);
    }
    return node;
    }
};

题目链接：617.合并二叉树

采用先序遍历：

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
    if(root1==nullptr) return root2;
    if(root2==nullptr) return root1;
    root1->val+=root2->val;
    root1->left=mergeTrees(root1->left,root2->left);
    root1->right=mergeTrees(root1->right,root2->right);
    return root1;

    }
};

题目链接：700.二叉搜索树中的搜索

采用递归法：

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root==nullptr||root->val==val) return root;
        if(root->left) return searchBST(root->left,val);
        if(root->right) return searchBST(root->right,val);
        return nullptr;

    }
};

题目链接：98.验证二叉搜索树

这道题可以利用中序遍历下，输出的二叉搜索树节点的数值是有序序列这点来作答

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    vector<int> result;
    void traversal(TreeNode* root)
    {   if(root==nullptr) return;
        traversal(root->left);
        result.push_back(root->val);
        traversal(root->right);
    }
public:
    bool isValidBST(TreeNode* root)
    {
        result.clear();
        traversal(root);
        for(int i=0;i<result.size()-1;i++)
        {
            if(result[i]>=result[i+1])
            return false;
        }
        return true;
    }
};