代码随想录第20天| 二叉树6
654. 最大二叉树(中等)
题目链接:654. 最大二叉树]
代码随想录:654. 最大二叉树
题目描述:给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
创建一个根节点,其值为 nums 中的最大值。
递归地在最大值 左边 的 子数组前缀上 构建左子树。
递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums 构建的 最大二叉树 。
class Solution {
public:
TreeNode * traversal(vector<int> &nums, int left, int right){
if(left >= right) return NULL;
int maxIndex = left;
int max = nums[left];
//找到最大值和下标
for(int i = left + 1; i < right; i++){
if(nums[i] > max){
maxIndex = i;
max = nums[i];
}
}
TreeNode * root = new TreeNode(max);
root->left = traversal(nums, left, maxIndex); // 保持左闭右开
root->right = traversal(nums, maxIndex + 1, right);
return root;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return traversal(nums, 0, nums.size());// 左闭右开
}
};
617.合并二叉树 (简单)
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
// 1在时
if(root1){
if(!root2) return root1; // 2不在 return 1
root1->val += root2->val;// 2在,计算val
}else //1不在,return 2
return root2;
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};
代码随想录:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2; // 如果t1为空,合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空,合并之后就应该是t1
// 修改了t1的数值和结构
t1->val += t2->val; // 中
t1->left = mergeTrees(t1->left, t2->left); // 左
t1->right = mergeTrees(t1->right, t2->right); // 右
return t1;
}
};
700.二叉搜索树中的搜索(简单)
LeetCode题目:700.二叉搜索树中的搜索
代码随想录:700.二叉搜索树中的搜索
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); //val比root小,左走
else
return searchBST(root->right, val);
}
};
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root == nullptr || root->val == val) return root;
//递归函数有返回值!!要用res接住
TreeNode * res = nullptr;
if(root->val > val)
res = searchBST(root->left, val); //val比root小,左走
else
res = searchBST(root->right, val);
return res;
}
};
98. 验证二叉搜索树(中等)☆
LeetCode题目:98. 验证二叉搜索树
代码随想录:98. 验证二叉搜索树
陷阱在于不能只比较左节点<中间<右节点
直观方法:二叉树中序遍历为数组,然后判断数组是否有序
递归:
用一个max记录中序遍历中当前节点之前的最大节点,以此判断是否有序
class Solution {
public:
long long max = LONG_MIN;
bool isValidBST(TreeNode* root) {
if(root == NULL) return true;
//中序遍历,更新遍历中的max,一旦发现max>=root,就返回false
bool left = isValidBST(root->left);
if(max < root->val) max = root->val;
else return false;
bool right = isValidBST(root->right);
return left && right;
}
};