LeetCode530 二叉搜索树的最小绝对差
题目链接:二叉搜索树的最小绝对差
思路
本题仍可使用中序遍历得出一个升序数组求得相邻最小绝对值差,但为了简便起见直接在遍历树时计算出差值,此时只需要一个指向前一个结点的指针pre。
代码
class Solution {
private:
int result = INT_MAX;
TreeNode* pre = NULL;
void traversal(TreeNode* cur) {
if (cur == NULL)
return;
traversal(cur->left); // 左
if (pre != NULL) { // 中
result = min(result, cur->val - pre->val);
}
pre = cur; // 记录前一个
traversal(cur->right); // 右
}
public:
int getMinimumDifference(TreeNode* root) {
traversal(root);
return result;
}
};
复杂度
时间复杂度:O(n)
空间复杂度:O(logn)
Leetcode501 二叉搜索树中的众数
题目链接:二叉搜索树中的众数
思路
若是普通的二叉树,可以遍历利用map统计。而本题是二叉搜索树,还可以使用中序遍历得到有序序列的特性,同上题的pre和cur指针。
代码
class Solution {
private:
int maxCount = 0; // 最大频率
int count = 0; // 统计频率
TreeNode* pre = NULL;
vector<int> result;
void searchBST(TreeNode* cur) {
if (cur == NULL)
return;
searchBST(cur->left); // 左
// 中
if (pre == NULL) { // 第一个节点
count = 1;
} else if (pre->val == cur->val) { // 与前一个节点数值相同
count++;
} else { // 与前一个节点数值不同
count = 1;
}
pre = cur; // 更新上一个节点
if (count == maxCount) { // 如果和最大值相同,放进result中
result.push_back(cur->val);
}
if (count > maxCount) { // 如果计数大于最大值频率
maxCount = count; // 更新最大频率
result
.clear(); // 很关键的一步,不要忘记清空result,之前result里的元素都失效了
result.push_back(cur->val);
}
searchBST(cur->right); // 右
return;
}
public:
vector<int> findMode(TreeNode* root) {
count = 0;
maxCount = 0;
pre = NULL; // 记录前一个节点
result.clear();
searchBST(root);
return result;
}
};
复杂度
时间复杂度:O(n)
空间复杂度:O(n)
Leetcode236 二叉树的最近公共祖先
题目链接:二叉树的最近公共祖先
思路
后序遍历是天然的回溯过程,可以根据左右子树的返回值,来处理中节点的逻辑。
要注意:
-
求最小公共祖先,需要从底向上遍历,那么二叉树,只能通过后序遍历(即:回溯)实现从底向上的遍历方式。
-
在回溯的过程中,必然要遍历整棵二叉树,即使已经找到结果了,依然要把其他节点遍历完,因为要使用递归函数的返回值(也就是代码中的left和right)做逻辑判断。
-
要理解如果返回值left为空,right不为空为什么要返回right,为什么可以用返回right传给上一层结果。
代码
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == q || root == p || root == NULL)
return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (left != NULL && right != NULL)
return root;
if (left == NULL && right != NULL)
return right;
else if (left != NULL && right == NULL)
return left;
else { // (left == NULL && right == NULL)
return NULL;
}
}
};
复杂度
时间复杂度:O(n)
空间复杂度:O(n)