最大二叉树
文档讲解:代码随想录
视频讲解: 又是构造二叉树,又有很多坑!| LeetCode:654.最大二叉树
状态
本题和之前通过中序后序构造前序类似,主要通过找到最大值为分割点,然后保证左闭右闭或者左闭右开,递归调用即可。
//获取最大值的下标函数
int GetMax(vector<int>& nums)
{
int m = 0;
for(int i= 1;i<nums.size();i++)
{
if(nums[m]<nums[i])
{
m = i;
}
}
return m;
}
获取根节点函数
TreeNode* GetRoot(vector<int>& nums)
{
if(nums.size() == 0) return nullptr;
//寻找树的父节点
int rootindex = GetMax(nums);
TreeNode* root = new TreeNode(nums[rootindex]);
//左数组
vector<int> left(nums.begin(),nums.begin()+rootindex);
//右数组
vector<int> right(nums.begin()+rootindex+1,nums.end());
root->left = GetRoot(left);
root->right = GetRoot(right);
return root;
}
可以发现,三道构造二叉树的题目都是通过前序遍历进行的。即先寻找到中节点(父节点)然后在利用递归为其左子节点和右子节点赋值。
合并二叉树
文档讲解:代码随想录
视频讲解: 一起操作两个二叉树?有点懵!| LeetCode:617.合并二叉树
状态:
本题看似遍历实则是构造+遍历,即最后会形成一颗新树。所以使用递归法较好解决。注意到递归的终止条件就是,当一棵树的节点为空,那么返回另外一个树此时的节点,相当于之后将该节点直接并入即可。单层递归就是求和运算,以及递归的调用
class Solution {
public:
TreeNode* SumTree(TreeNode* root1, TreeNode* root2)
{
//即哪一个先走完,返回的是另一个剩下的
if(root1 == nullptr) return root2;
if(root2 == nullptr) return root1;
//直接在root1树上修改
//中节点处理逻辑
root1->val += root2->val;
//左节点和右节点递归
root1->left = SumTree(root1->left,root2->left);
root1->right = SumTree(root1->right,root2->right);
return root1;
}
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
return SumTree(root1,root2);
}
};
当然迭代也是可以做的,就是利用一个队列同时维护两个树的传入,我们可以定义一个新的返回节点,也可以使用root1或者root2。
注意我们队列中的元素只是需要加法的元素,如果不是那么没必要入队,可以直接将root2的节点赋值过去即可
root->val += root2->val;
if(root1->left && root2->left) push
if(root1->right && root2->right) push
if(!root1->left && root2->left) root1->left = root2->left;
if(!root1->right && root2->right) root1->right = root2->right;
二叉搜索树中的搜索
文档讲解:代码随想录
视频讲解: 不愧是搜索树,这次搜索有方向了!| LeetCode:700.二叉搜索树中的搜索
状态
- 递归
BST左子节点比根节点小,右子节点比根节点大
- 返回条件
当root为空,那么返回空,如果root为当前val返回root
if(root && root->val == val) return root;
if(root == nullptr) return nullptr;
- 单层递归
当当前父节点的值小于val,递归右子树
当当前父节点的值大于val,递归左子树
中节点 处理返回过程
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root && root->val == val) return root;
if(root == nullptr) return nullptr;
if(root->val < val)
{
root = searchBST(root->right,val);
}
else
{
root = searchBST(root->left,val);
}
return root;
}
};
验证二叉搜索树
文档讲解:代码随想录
视频讲解: 你对二叉搜索树了解的还不够! | LeetCode:98.验证二叉搜索树
状态
二叉搜索树要善于运用中序遍历,这样得到的就是左中右,是一个升序的数组
同样在判断的时候左中右,然后更新最大值,如果是二叉搜索树,那么最大值会一直被更新到最后一个节点。
递归函数的返回值通常在寻找某条边或者某个节点是需要定义,其余情况可以不用定义。
那么对于这道题,左子树返回一个迭代bool,右子树返回一个迭代bool,中节点用来处理当前子树的满足情况。
class Solution {
public:
TreeNode* pre = nullptr;
bool isValidBST(TreeNode* root) {
if(root == nullptr) return true;
//左子树迭代
bool left = isValidBST(root->left);
//中节点处理
if(pre != nullptr && pre->val >= root->val) return false;
//中序遍历,不断更新节点
pre = root;
//右子树迭代
bool right = isValidBST(root->right);
//当前父节点的子树满足情况
return left&&right;
}
};
本题要注意,二叉搜索树是左子树上的值要小于父节点,右子树上的值要大于父节点