算法训练营 第二十天

算法训练营 第二十天

第19天休息日，今天是二叉树day06~

654.最大二叉树

这道题也是构造二叉树的题，和昨天的已知前序中序构造二叉树思路一样

class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        if (nums.size() == 0)
            return nullptr;
        if (nums.size() == 1)
            return new TreeNode(nums.front());
        
        int index = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[index] < nums[i])
                index = i;
        }
        TreeNode* root = new TreeNode(nums[index]);
        vector<int> leftVec(nums.begin(), nums.begin() + index);
        vector<int> rightVec(nums.begin() + index + 1, nums.end());
        root->left = constructMaximumBinaryTree(leftVec);
        root->right = constructMaximumBinaryTree(rightVec);
        return root;
            
    }
};

代码随想录中的方法用双指针的方法，避免了构造vector<int> leftVec和vector<int> rightVec，内存损耗更少。

class Solution {
private:
    // 在左闭右开区间[left, right)，构造二叉树
    TreeNode* traversal(vector<int>& nums, int left, int right) {
        if (left >= right) return nullptr;

        // 分割点下标：maxValueIndex
        int maxValueIndex = left;
        for (int i = left + 1; i < right; ++i) {
            if (nums[i] > nums[maxValueIndex]) maxValueIndex = i;
        }

        TreeNode* root = new TreeNode(nums[maxValueIndex]);

        // 左闭右开：[left, maxValueIndex)
        root->left = traversal(nums, left, maxValueIndex);

        // 左闭右开：[maxValueIndex + 1, right)
        root->right = traversal(nums, maxValueIndex + 1, right);

        return root;
    }
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        return traversal(nums, 0, nums.size());
    }
};

617.合并二叉树

这道题用递归的思路，三步走写下去就行，没什么难度

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

    }
};

700.二叉搜索树中的搜索

二叉搜索树的结点有了顺序，找起来就方便了很多

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

    }
};

98.验证二叉搜索树

这道题得用中序遍历，二叉搜索树在中序遍历的时候是一个有序数组。因此，只用判断前一个结点的值是否比有一个小就行。

class Solution{
public:
    bool isValidBST(TreeNode * root) {
        if (root == nullptr) {
            return true;
        }
        bool left = isValidBST(root->left);
        if (pre&&pre->val >= root->val)
            return false;
        pre = root;
        bool right = isValidBST(root->right);
        return left && right;
    }

private:
    TreeNode* pre = nullptr;
};