题目大意:给出的二叉搜索树是含有重复结点的,找出树中出现最频繁的数字,如果有多个出现次数都为最大值的顺序无所谓
分析:dfs。由于二叉搜索树的特点,中序遍历就会得到非降序序列,相同节点是连续的,所以只需要中序遍历二叉搜索树并不断记录每个数字的出现次数,更新最大频率数字即可。
代码:
class Solution {
public:
vector<int> findMode(TreeNode* root) {
inorder(root);
return result;
}
private:
vector<int> result;
int maxCount = 0, currentVal, tempCount = 0;
void inorder(TreeNode* root) {
if (root == NULL) return;
inorder(root->left);
if (root->val != currentVal) {
currentVal = root->val;
tempCount = 0;
}
tempCount++;
if (tempCount > maxCount) { //出现频率更大,更新答案数组
maxCount = tempCount;
result.clear();
result.push_back(root->val);
}
else if (tempCount == maxCount) { //多个结点的频率都为最大频率
result.push_back(root->val);
}
inorder(root->right);
}
};