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

654.最大二叉树

 这题有点像昨天的中序遍历和后序遍历得出前序遍历的题，都是拆分成两个数组分别进行递归。

/**
 * 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) {
        auto maxIt = max_element(nums.begin(), nums.end());
        TreeNode* root = new TreeNode(*maxIt);
        vector<int> leftNums(nums.begin(), maxIt);
        vector<int> rightNums(maxIt+1,nums.end());
        if(!leftNums.empty()){ 
            root->left = constructMaximumBinaryTree(leftNums);
        }
        if(!rightNums.empty()){
            root->right = constructMaximumBinaryTree(rightNums);    
        }
        return root;
    }
};

617.合并二叉树

        根据合并的规则可以写出不同的条件对应情况，再采用前序遍历，最后返回根节点。

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        TreeNode* root = new TreeNode(0);
        if(root1 && root2) root->val = root1->val + root2->val;      
        if(!root1 && root2) return root2;
        if(root1 && !root2) return root1;
        if(!root1 && !root2) root = nullptr;
        if(root) {
            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->val == val) {
            return root;
        }        
        if(root->val < val && root->right) {
            return searchBST(root->right,val);
        }
        if(root->val > val && root->left) {
            return searchBST(root->left,val);
        }

        return nullptr;

    }
};

 

98.验证二叉搜索树

        中序遍历二叉搜索树会是从小到大的顺序，所以将其值保存到一个数组中再判断是否按从小到大顺序排列。

class Solution {
public:
    vector<int> result;
    bool isValidBST(TreeNode* root) {
        if(!root) return true;
        auto traversal = [&](auto& self, TreeNode* node) -> void {
            if (node->left) self(self, node->left);
            result.push_back(node->val);
            if (node->right) self(self, node->right);
        };
        traversal(traversal, root);
        for(int i = 0; i < result.size() - 1;++i) {
            if(result[i] >= result[i + 1]) return false;
        }
        return true;
    }
};