LeetCode 530. 二叉搜索树的最小绝对差
思路:
寻找二叉搜索树的最小绝对差,可以合理利用二叉树的特性,也就是中序遍历二叉搜索树的时候,其结果是一个升序数组,因为二叉搜索树的节点大小关系为左节点小于中间节点,右节点大于中间节点,所以按照左->中->右的顺序遍历即可得到一个升序数组。
利用这个特性,很容易想到的一个办法就是先中序遍历二叉树得到一个数组,再遍历这个数组并记录下来每相邻的两个元素的最小差值。但实际上根本没必要这么做,因为在遍历二叉树的时候就可以完成比较的工作,用一个变量记录最小绝对差,然后每次比较的时候都更新这个变量。
这个办法就是双指针法,为了每次遍历节点的时候都能和前一个节点计算差值,需要一个额外的指针记录前一个节点,在中序遍历的时候,我们每次都先处理prev和cur节点,然后取两个节点的最小差值,再更新cur和prev节点,令prev=cur,cur=cur->right。
// 处理中间节点
// 取两个节点的最小差值
if (prev != nullptr)
minDiff = min(minDiff, cur->val - prev->val);
// 令prev等于当前节点
prev = cur;
这种双指针的办法可以使我们遍历二叉树的时候同时处理两个节点,一个为当前遍历的节点,一个为前一个遍历过的节点,这样可以节省遍历数组所花的时间和空间。
代码:
class Solution {
public:
int minDiff = INT_MAX;
TreeNode* prev = nullptr;
int getMinimumDifference(TreeNode* root) {
inorder(root);
return minDiff;
}
void inorder(TreeNode* cur)
{
if (cur == nullptr)
return;
// 中序遍历,先遍历左边的节点
inorder(cur->left);
// 处理中间节点
// 取两个节点的最小差值
if (prev != nullptr)
minDiff = min(minDiff, cur->val - prev->val);
// 令prev等于当前节点
prev = cur;
// 遍历右边的节点
inorder(cur->right);
return;
}
};
LeetCode 501.二叉搜索树中的众数
思路:
注意二叉搜素数一般是没有重复节点的,但是这道题中的二叉树的元素可以重复。这道题的思路和530. 二叉搜索树的最小绝对差一致,都是用中序遍历的方法,然后用一个prev节点储存之前的节点方便和当前节点作比较。众数为二叉树中出现次数最多的数,因此在中序遍历的过程中,如果一个数出现了很多次,那么prev和cur节点的值一定会出现很多次相同的情况,因此只需要把prev和cur相同的情况的次数count记录下来,就知道这个数重复出现了多少次。
当prev和cur值相同时,count++,当不同时,count就重置为1,表示这个数只出现过一次。同时,我们需要另一个变量maxCount用于记录当前重复最多次数的数的重复次数是多少。众数可以是很多个,因此当发现了maxCount等于count的时候,就把当前数push到结果里,如果发现了maxCount小于count,则说明当前的数重复次数多于之前记录的次数最大值,那么当前的数便成为了新的众数,需要把之前记录在结果里的数全部clear,再把新的众数放进结果里。
代码:
class Solution {
public:
int count = 1;
vector<int> ans;
TreeNode *prev;
int maxCount = 1;
vector<int> findMode(TreeNode* root) {
inorder(root);
return ans;
}
void inorder(TreeNode* root)
{
if (root == nullptr)
return;
inorder(root->left);
if (prev != nullptr && prev->val == root->val)
// 如果现节点和之前节点值一样,数量加1
count++;
else
// 如果不一样,重置数量
count = 1;
if (maxCount == count)
// 如果当前节点出现频率等于最大频率,说明有多个众数
ans.push_back(root->val);
else if (maxCount < count)
{
// 如果当前元素出现频率大于最大频率,更新最大频率
maxCount = count;
// 清除数组里的元素
ans.clear();
// 把新的最大频率的元素添加进来
ans.push_back(root->val);
}
// 更新之前的节点
prev = root;
inorder(root->right);
return;
}
};
LeetCode 236. 二叉树的最近公共祖先
思路:
这道题和235. 二叉搜索树的最近公共祖先类似,但是解题过程却大不相同。这里采用后序遍历的方法,从二叉树的底端自底向上查找,分别用left和right两个节点表示查找到的p、q两个节点或者分别包含p、q两个节点的子树的根节点,判断方法为当root找到了p或者q的其中一个节点,就把这个节点返回给上一个父节点,如果没找到,则返回null。这样就相当于把p、q两个节点向上传递,直到这两个节点分别填满了left和right两个节点,因为root节点为left和right的父节点,这就表示root节点也是p和q的公共祖先。
为什么root一定是p和q的最近公共祖先呢?可以这样想,当left或者right其中一个为null的时候,这就表示了,root作为left或者right节点的祖先,如果left不为null,那么left一定是由p或者q的任意一个节点传递上来的,所以root一定也是p或者q其中之一的祖先,但是这个时候right为null,那么root一定不是p或者q中的另一个的祖先,要同时满足root成为两者的祖先,一定是left和right同时不为null的情况,所以在刚好满足这种情况的时候,root为第一个p和q的公共祖先,也就是最近公共祖先。
代码:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL)
return NULL;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (root->val == p->val || root->val == q->val)
return root;
if (left != NULL && right != NULL)
return root;
else if (left == NULL && right != NULL)
return right;
else if (left != NULL && right == NULL)
return left;
else
return NULL;
}
};