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

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

654.最大二叉树

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

解题思路

构造一棵树通常使用的是前序遍历，因为需要先构造中间节点，再构造左子树和右子树。

递归法三步：

• 确定递归函数参数和返回值：传入参数为存放元素的数组，返回值是该数组构造的二叉树的头结点，返回类型是指向节点的指针。
• 终止条件：若传入的数组大小为1，说明遍历到了叶子节点
• 单层递归的逻辑：1.找到数组中最大的值和对应的下标， 最大的值构造根节点，下标用来下一步分割数组。2.最大值所在的下标左区间 构造左子树。3.最大值所在的下标右区间 构造右子树。

代码实现

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> vec(nums.begin(), nums.begin() + maxValueIndex);
            node->left = constructMaximumBinaryTree(vec);
        }
        if (maxValueIndex < (nums.size() - 1)) {
            vector<int> vec(nums.begin() + maxValueIndex + 1, nums.end());
            node->right = constructMaximumBinaryTree(vec);
        }
        return node;
    }
};

题目总结

注意类似用数组构造二叉树的题目，每次分隔尽量不要定义新的数组，而是通过下标索引直接在原数组上操作，这样可以节约时间和空间上的开销。

617.合并二叉树

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

解题思路

考虑使用递归法解题

• 确定递归函数的参数和返回值：参数是两个二叉树的根节点，返回值是合并之后二叉树的根节点
• 确定终止条件：判断两棵树遍历的节点
• 确定单层递归的逻辑：将两棵树的元素加到一起

代码实现

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;
    }
};

题目总结

将 root1这棵树重复利用，root1 就是合并之后树的根节点

700.二叉搜索树中的搜索

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

解题思路

首先需要了解二叉搜索树的的特性：

• 若它的左子树不为空，则左子树上所有节点的值均小于它根节点的值
• 若它的右子树不为空，则右子树上所有节点的值均大于它根节点的值
• 它的左右子树也分别为二叉搜索树

知道了特性后，本题使用递归法来求解

• 递归函数的参数就是传入的根节点和需要搜索的数值，返回值是要搜索的数值所在的节点
• 终止条件是当 root 为空时，或者找到这个数值，返回 root 节点
• 单层递归逻辑：二叉搜索树的节点是有序的，所以可以有方向的搜索，若 root->val 大于 val，搜索左子树，若 root->val 小于 val，搜索右子树

代码实现

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

题目总结

基于二叉搜索树的有序性，遍历的时候会比普通二叉树简单些。

98.验证二叉搜索树

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

解题思路

使用递归法将二叉搜索树转变成一个数组，比较这个数组是不是有序的，若有序，则说明是二叉搜索树。

代码实现

class Solution {
public:
    vector<int> vec;
    void taversal(TreeNode* root) {
        if (root == nullptr)
            return;
        taversal(root->left);
        vec.push_back(root->val);
        taversal(root->right);
    }
    bool isValidBST(TreeNode* root) {
        vec.clear();
        taversal(root);
        for (int i = 1; i < vec.size(); i++) {
            if (vec[i] <= vec[i - 1]) {
                return false;
            }
        }
        return true;
    }
};

题目总结

将二叉树转变为线性序列来判断其是否为二叉搜索树。