给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。
假定 BST 有如下定义:
结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:
给定 BST [1,null,2,2],
1\2/2
返回[2].
提示:如果众数超过1个,不需考虑输出顺序
代码如下:
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:int base = 0;int cnt = 0;int maxcnt = 0;vector<int>ans;vector<int> findMode(TreeNode* root) {dfs(root);return ans;}void compare_node(int x){if (x==base){cnt++;}else{base = x;cnt = 1;}if (cnt == maxcnt) ans.push_back(base);if (cnt > maxcnt){maxcnt = cnt;ans = vector<int>{base};}}void dfs(TreeNode *root){if (root==nullptr) return ;dfs(root->left);compare_node(root->val);dfs(root->right);}};