1、二叉搜索树的最小绝对差
最直观的想法,中序递归:
class Solution {
private:
vector<int> vec;
void traversal(TreeNode* root) { //中序递归
if (root == nullptr) return;
traversal(root->left);
vec.push_back(root->val); // 将二叉搜索树转换为有序数组
traversal(root->right);
}
public:
int getMinimumDifference(TreeNode* root) {
traversal(root); //因题目规定节点数最小为2,所以不再判断小于2
int result = INT_MAX; //题目中最大差值为10^5,所以INT_MAX足够大了(2^31-1)
for (int i = 1; i < vec.size(); ++i) { // 统计有序数组的最小差值
result = min(result, vec[i] - vec[i-1]);
}
return result;
}
};
双指针:
class Solution {
private:
int result = INT_MAX;
TreeNode* pre = nullptr;
void traversal(TreeNode* cur) {
if (cur == nullptr) return;
traversal(cur->left); // 左
if (pre != nullptr){ // 中 (避免下面操作空指针)
result = min(result, cur->val - pre->val);
}
pre = cur; // 记录前一个
traversal(cur->right); // 右
}
public:
int getMinimumDifference(TreeNode* root) {
traversal(root);
return result;
}
};
中序迭代:
class Solution {
public:
int getMinimumDifference(TreeNode* root) {
stack<TreeNode*> st;
TreeNode* cur = root;
TreeNode* pre = nullptr;
int result = INT_MAX;
while (cur != nullptr || !st.empty()) {
if (cur != nullptr) { // 指针来访问节点,访问到最底层
st.push(cur); // 将访问的节点放进栈
cur = cur->left; // 左
} else {
cur = st.top();
st.pop();
if (pre != nullptr) { // 中
result = min(result, cur->val - pre->val);
}
pre = cur;
cur = cur->right; // 右
}
}
return result;
}
};
2、二叉搜索树中的众数
如果是一般二叉树,可以这么做
class Solution {
private:
void searchBST(TreeNode* cur, unordered_map<int, int>& map) { // 前序遍历
if (cur == nullptr) return ;
map[cur->val]++; // 统计元素频率 key无序且不重复
searchBST(cur->left, map);
searchBST(cur->right, map);
}
bool static cmp(const pair<int, int>& a, const pair<int, int>& b) {
return a.second > b.second;
}
public:
vector<int> findMode(TreeNode* root) {
unordered_map<int, int> map; // key:元素,value:元素频率
vector<int> result; //结果数组
if (root == nullptr) return result;
searchBST(root, map);
vector<pair<int, int>> vec(map.begin(), map.end()); //必须复制到vector容器中才能进行排序
sort(vec.begin(), vec.end(), cmp); // 从大到小给频率排个序
result.push_back(vec[0].first);
for (int i = 1; i < vec.size(); ++i) {
// 取最高的放到result数组中
if (vec[i].second == vec[0].second)
result.push_back(vec[i].first);
else break;
}
return result;
}
};
本题是二叉搜索树,所以可以利用二叉搜索树的性质,进行简化
class Solution {
private:
vector<int> result; //结果数组
TreeNode* pre = nullptr; //前指针
int count = 1; //计数器
int maxCount = 1;
void inorder(TreeNode* cur) {
if (cur == nullptr) return; //终止条件
inorder(cur->left); //左
//中
if (pre && pre->val == cur->val) { //注意这里要对pre进行判空
count++;
} else {
count = 1;
}
pre = cur;
if (count == maxCount) {
result.push_back(cur->val);
} else if (count > maxCount) {
maxCount = count;
result.clear();
result.push_back(cur->val);
}
inorder(cur->right); //右
}
public:
vector<int> findMode(TreeNode* root) {
inorder(root);
return result;
}
};
中序迭代:
class Solution {
public:
vector<int> findMode(TreeNode* root) {
stack<TreeNode*> st;
TreeNode* cur = root;
TreeNode* pre = nullptr;
int maxCount = 1; // 最大频率
int count = 1; // 统计频率
vector<int> result;
while (cur != nullptr || !st.empty()) {
if (cur != nullptr) { // 指针来访问节点,访问到最底层
st.push(cur); // 将访问的节点放进栈
cur = cur->left; // 左
} else {
cur = st.top();
st.pop(); // 中
if (pre && pre->val == cur->val) { // 与前一个节点数值相同
count++;
} else { // 与前一个节点数值不同
count = 1;
}
if (count == maxCount) { // 如果和最大值相同,放进result中
result.push_back(cur->val);
} else if (count > maxCount) { // 如果计数大于最大值频率
maxCount = count; // 更新最大频率
result.clear(); // 很关键的一步,不要忘记清空result,之前result里的元素都失效了
result.push_back(cur->val);
}
pre = cur;
cur = cur->right; // 右
}
}
return result;
}
};
其实主体部分还是一样的。
3、二叉树的最近公共祖先
后序递归:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == q || root == p || root == nullptr) return root; //终止条件
TreeNode* left = lowestCommonAncestor(root->left, p, q); //左
TreeNode* right = lowestCommonAncestor(root->right, p, q); //右
if (left != nullptr && right != nullptr) return root;
else if (left == nullptr && right != nullptr) return right;
else if (left != nullptr && right == nullptr) return left;
else { // (left == NULL && right == NULL)
return nullptr;
}
}
};
可以精简为:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == q || root == p || root == nullptr) return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (left != nullptr && right != nullptr) return root;
else if (left == nullptr) return right;
else return left;
}
};
本题使用了递归的回溯,因此不适合使用迭代法去求解。