给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
思路
递归法
既然是搜索树,中序遍历就是有序数组。
如何不利用数组,直接在树上进行操作呢?
弄一个指针指向前一个节点,这样每次cur(当前节点)才能和pre(前一个节点)作比较。
而且初始化的时候pre = NULL,这样当pre为NULL时候,我们就知道这是比较的第一个元素。
if (pre == NULL) { // 第一个节点count = 1; // 频率为1
} else if (pre->val == cur->val) { // 与前一个节点数值相同count++;
} else { // 与前一个节点数值不同count = 1;
}
pre = cur; // 更新上一个节点
但是由于树中的众数可能存在多个,我们需要用数组存放多个元素。
设置频率count和最大频率maxCount。
if (count > maxCount) { // 如果计数大于最大值maxCount = count; // 更新最大频率result.clear(); // 很关键的一步,不要忘记清空result,之前result里的元素都失效了result.push_back(cur->val);
}
完整代码:
class Solution {
public: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.push_back(cur->val);}if(count > maxCount){maxCount = count;result.clear();result.push_back(cur->val);}searchBST(cur->right);return;}vector<int> findMode(TreeNode* root) {//count = 0;//maxCount = 0;//TreeNode* pre = NULL;result.clear();searchBST(root);return result;}
};
迭代法
将中序遍历转为迭代,中间节点的处理逻辑一致。
class Solution {
public:vector<int> findMode(TreeNode* root) {stack<TreeNode*> st;TreeNode* cur = root;TreeNode* pre = NULL;int maxCount = 0;int count = 0;vector<int> result;while(cur != NULL || !st.empty()){if(cur != NULL){st.push(cur);cur = cur->left;}else{cur = st.top();st.pop();if(pre == NULL) count = 1;else if(pre->val == cur->val) count++;else count = 1;if(count == maxCount){result.push_back(cur->val);}if(count > maxCount){maxCount = count;result.clear();result.push_back(cur->val);}pre = cur;cur = cur->right; }}return result;}
};