代码随想录算法训练营第17天 | 654.最大二叉树、617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树
文章目录
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.合并二叉树
解题思路
考虑使用递归法解题
- 确定递归函数的参数和返回值:参数是两个二叉树的根节点,返回值是合并之后二叉树的根节点
- 确定终止条件:判断两棵树遍历的节点
- 确定单层递归的逻辑:将两棵树的元素加到一起
代码实现
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.二叉搜索树中的搜索
解题思路
首先需要了解二叉搜索树的的特性:
- 若它的左子树不为空,则左子树上所有节点的值均小于它根节点的值
- 若它的右子树不为空,则右子树上所有节点的值均大于它根节点的值
- 它的左右子树也分别为二叉搜索树
知道了特性后,本题使用递归法来求解
- 递归函数的参数就是传入的根节点和需要搜索的数值,返回值是要搜索的数值所在的节点
- 终止条件是当 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.验证二叉搜索树
解题思路
使用递归法将二叉搜索树转变成一个数组,比较这个数组是不是有序的,若有序,则说明是二叉搜索树。
代码实现
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;
}
};
题目总结
将二叉树转变为线性序列来判断其是否为二叉搜索树。