代码随想录算法训练营第二十天| LeetCode654.最大二叉树、LeetCode700.二叉搜索树中的搜索、LeetCode617.合并二叉树、LeetCode98.验证二叉搜索树
2023年3月12日
第二十六天补,最近一直在补笔记,顺序可能有点乱,要抓紧跟上呀
文章目录
654. 最大二叉树
题目链接:654. 最大二叉树
思路:
- 在数组中找出最大的值作为根节点,该值左侧的子数组中的最大值作为它的左孩子节点,该值右侧的子数组中的最大值作为它的右孩子节点,
- 这样的逻辑可以很方便的使用递归来表现
时间复杂度:O(n)
,空间复杂度O(n)
代码:
class Solution
{
public:
TreeNode *constructMaximumBinaryTree(vector<int> &nums)
{
// 如果数组大小为0,则给上一个节点附上空指针,还能避免下方数组索引越界
if (nums.size() == 0)return nullptr;
// 找出最大值的值和位置
int max = nums[0];
int pos = 0;
for (int i = 1; i < nums.size(); ++i)
{
if (nums[i] > max)
{
max = nums[i];
pos = i;
}
}
// 构造当前节点
TreeNode *root = new TreeNode(max);
// 生成两个子数组并递归
std::vector<int> left(nums.begin(), nums.begin() + pos);
std::vector<int> right(nums.begin() + pos + 1, nums.end());
root->left = constructMaximumBinaryTree(left);
root->right = constructMaximumBinaryTree(right);
// 返回当前节点
return root;
}
};
总结:
- 这道题只要理清楚了逻辑,实现起来并不困那
- 做题体验:
哈哈
(哈哈
>嘿嘿
>哎哎
>嘤嘤
) - 时间:
未计
一刷
617. 合并二叉树
题目链接:617. 合并二叉树
思路:
- 这道题要用递归同时遍历两个二叉树,判断返回的条件是其中一个关键
- 这里我写两版代码,一版是构造新二叉树,一版是在原先的二叉树上修改
新二叉树
时间复杂度:O(n)
,空间复杂度O(n)
代码:
class Solution
{
public:
TreeNode *mergeTrees(TreeNode *root1, TreeNode *root2)
{
// 当前构造的节点
TreeNode *root = new TreeNode;
// 分别判断四种情况
if (root1 == nullptr && root2 == nullptr)
return nullptr;
else if (root1 == nullptr && root2 != nullptr)
{
root->val = root2->val;
// 用前序遍历的顺序构造这棵二叉树
root->left = mergeTrees(nullptr, root2->left);
root->right = mergeTrees(nullptr, root2->right);
}
else if (root1 != nullptr && root2 == nullptr)
{
root->val = root1->val;
root->left = mergeTrees(root1->left, nullptr);
root->right = mergeTrees(root1->right, nullptr);
}
else
{
root->val = root1->val + root2->val;
root->left = mergeTrees(root1->left, root2->left);
root->right = mergeTrees(root1->right, root2->right);
}
return root;
}
};
原地修改
时间复杂度:O(n)
,空间复杂度O(1)
代码:
class Solution
{
public:
TreeNode *mergeTrees(TreeNode *root1, TreeNode *root2)
{
// 如果root1为空,就把root2接过来,如果root2为空,就把root1接过来
if (root2 == nullptr)return root1;
if (root1 == nullptr)return root2;
// 在root1的基础上做修改
root1->val = root1->val + root2->val;
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};
总结:
- 这道题不难,只要理清遍历的逻辑就可以解出
- 做题体验:
哈哈
(哈哈
>嘿嘿
>哎哎
>嘤嘤
) - 时间:
未计
一刷
700. 二叉搜索树中的搜索
题目链接:700. 二叉搜索树中的搜索
思路:
- 一个简单的二叉搜索树的使用
时间复杂度:O(n)
,空间复杂度O(1)
代码:
class Solution
{
public:
TreeNode *searchBST(TreeNode *root, int val)
{
// 再遇到空指针和找到的情况,给result赋值
if (root == nullptr)return result;
if (root->val == val)result = root;
else if (root->val < val) searchBST(root->right, val);
else searchBST(root->left, val);
return result;
}
TreeNode *result = nullptr;
};
总结:
- 比较简单
- 做题体验:
哈哈
(哈哈
>嘿嘿
>哎哎
>嘤嘤
) - 时间:
未计
一刷
98. 验证二叉搜索树
题目链接:98. 验证二叉搜索树
思路:
-
对于一个二叉搜索树,首先我们必须明确地是,二叉搜索树的中序遍历结果,就是一个单调递增数组
-
所以我们只要求得二叉树的中序遍历数组,查看是否是单调递增就可以,这是
思路1
-
思路2
在中序遍历二叉树的时候其实就相当于遍历了他的中序遍历数组,只要用一个变量记录前一个值,再进行比较就可以了,可以节省时间复杂度 -
思路3
和思路2接近,用一个指针来记录上一个元素
思路1:数组记录
时间复杂度:O(n)
,空间复杂度O(n)
代码:
// 序列化方式
class Solution
{
public:
bool isValidBST(TreeNode *root)
{
std::vector<int> valVct;
travelNode(root,valVct);
int pre = valVct[0];
for(int i = 1;i<valVct.size();++i)
{
if(valVct[i] <= pre)return false;
else pre= valVct[i];
}
return true;
}
// 中年序遍历
void travelNode(TreeNode* node,vector<int> &valVct)
{
if(node == nullptr)return;
travelNode(node->left,valVct);
valVct.push_back(node->val);
travelNode(node->right,valVct);
}
};
思路2:整型记录
时间复杂度:O(n)
,空间复杂度O(1)
代码:
// 用最大值变量
class Solution
{
public:
bool isValidBST(TreeNode *root)
{
if (root == nullptr)
return true;
// 用左中右的顺序遍历二叉树,相当于对中序遍历数组进行判断
auto left = isValidBST(root->left);
if (root->val <= max)
return false;
else
{
// 每次遍历更新最大值
max = root->val;
}
auto right = isValidBST(root->right);
return left && right;
}
long long max = INT64_MIN;
};
思路3:指针记录
时间复杂度:O(n)
,空间复杂度O(1)
代码:
// 双指针方式
class Solution
{
public:
bool isValidBST(TreeNode *root)
{
if (root == nullptr)
return true;
auto left = isValidBST(root->left);
if (pre == nullptr)
pre = root;
else
{
if (root->val <= pre->val)
return false;
else
{
pre = root;
}
}
auto right = isValidBST(root->right);
return left && right;
}
TreeNode *pre = nullptr;
};
总结:
*
- 做题体验:
嘤嘤
(哈哈
>嘿嘿
>哎哎
>嘤嘤
) - 时间:
未计
一刷