一、最大二叉树
1.题目:给定一个不重复的整数数组 nums
。 最大二叉树 可以用下面的算法从 nums
递归地构建:
- 创建一个根节点,其值为
nums
中的最大值。 - 递归地在最大值 左边 的 子数组前缀上 构建左子树。
- 递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums
构建的 最大二叉树 。
2.思路:
3.注意:
(1)方法一中在递归时不断创建新数组,会消耗大量内存,修改为方法二,使用记录下标的方式在原数组 nums 上获取数值即可。方法二中局部变量甚至不需要 maxValue ,使用 nums[maxIndex] 即可获取所需数值,无需用变量记录。
(2)方法二中在寻找区间最大值时把变量定义和 for循环参数改成注释部分会运行超时。
4.代码:
方法一:
/**
* 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) {
int maxValue=0,maxIndex=0;
for(int i=0; i<nums.size(); i++)
{
if(nums[i]>maxValue)
{
maxValue = nums[i];
maxIndex = i;
}
}
TreeNode* root = new TreeNode(maxValue);
vector<int> leftNums;
vector<int> rightNums;
for(int i=0; i<maxIndex; i++)
{
leftNums.push_back(nums[i]);
}
for(int i=maxIndex+1; i<nums.size(); i++)
{
rightNums.push_back(nums[i]);
}
if(!leftNums.empty())
root->left = constructMaximumBinaryTree(leftNums);
if(!rightNums.empty())
root->right = constructMaximumBinaryTree(rightNums);
return root;
}
};
方法二:
/**
* 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* traversal(vector<int> nums, int leftIndex, int rightIndex)
{
if( leftIndex >= rightIndex) return NULL;
int maxValue=nums[leftIndex],maxIndex=leftIndex;
for(int i=leftIndex+1; i<rightIndex; i++)
// int maxValue=0,maxIndex=0;
// for(int i=leftIndex; i<rightIndex; i++)
{
if(nums[i]>maxValue)
{
maxValue = nums[i];
maxIndex = i;
}
}
TreeNode* root = new TreeNode(maxValue);
root->left = traversal(nums, leftIndex, maxIndex);
root->right = traversal(nums, maxIndex+1, rightIndex);
return root;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return traversal(nums, 0, nums.size());
}
};
二、合并二叉树
1.题目:给你两棵二叉树: root1
和 root2
。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
2.思路:
3.注意:
4.代码:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1 == NULL) return root2;
if(root2 == NULL) return root1;
root1->val += root2->val;
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};
三、二叉搜索树中的搜索
1.题目:给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL。
2.思路:
3.注意:
4.代码:
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root == NULL) return NULL;
if(root->val == val) return root;
if(root->val > val)
return searchBST(root->left, val);
if(root->val <val)
return searchBST(root->right, val);
return NULL;
}
};
四、验证二叉搜索
1.题目:给你一个二叉树的根节点 root
,判断其是否是一个有效的二叉搜索树。有效二叉搜索树定义如下:
- 节点的左子树只包含 小于 当前节点的数。
- 节点的右子树只包含 大于 当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
2.思路:先用递归把二叉树变为有序数组,然后检查数组是否有序。
3.注意:
(1)不能单纯的比较左节点小于中间节点,右节点大于中间节点就完事了。
示例:
(2)样例中最小节点 可能是int的最小值,如果这样使用最小的int来比较也是不行的。此时可以初始化比较元素为longlong的最小值。
问题可以进一步演进:如果样例中根节点的val 可能是longlong的最小值 又要怎么办呢?
如方法二所示,取最左面节点的数值来比较。
4.代码:
方法一:
class Solution {
private:
vector<int> vec;
void traversal(TreeNode* root) {
if (root == NULL) return;
traversal(root->left);
vec.push_back(root->val); // 将二叉搜索树转换为有序数组
traversal(root->right);
}
public:
bool isValidBST(TreeNode* root) {
vec.clear(); // 不加这句在leetcode上也可以过,但最好加上
traversal(root);
for (int i = 1; i < vec.size(); i++) {
// 注意要小于等于,搜索树里不能有相同元素
if (vec[i] <= vec[i - 1]) return false;
}
return true;
}
};
方法二:
class Solution {
public:
TreeNode* pre = NULL; // 用来记录前一个节点
bool isValidBST(TreeNode* root) {
if (root == NULL) return true;
bool left = isValidBST(root->left);
if (pre != NULL && pre->val >= root->val) return false;
pre = root; // 记录前一个节点
bool right = isValidBST(root->right);
return left && right;
}
};