文章目录
- 题目
- 方法一:暴力哈希
- 方法二:利用二叉搜索树的特性(递归+双指针)
题目
方法一:暴力哈希
这是针对于普通二叉树的解法 统计number出现次数 然后将次数最大的众数集 取出来
Map<Integer , Integer > map = new HashMap<>();PriorityQueue<int[]> priori = new PriorityQueue<>((a,b)->b[1]-a[1]);//优先队列 按数组第二个元素从大到小排List<Integer> list = new ArrayList<>();public int[] findMode(TreeNode root) {dfs(root);for(Map.Entry<Integer, Integer> num : map.entrySet()) priori.offer(new int[]{num.getKey(),num.getValue()});int max = priori.peek()[1];int size =priori.size();for(int i = 0 ; i<size ; i++){int [] a1 = priori.poll();if(max == a1[1]) list.add(a1[0]);} return list.stream().mapToInt(Integer::intValue).toArray(); }public void dfs(TreeNode root){if(root == null) return;map.put(root.val,map.getOrDefault(root.val, 0)+1);dfs(root.left);dfs(root.right);}
方法二:利用二叉搜索树的特性(递归+双指针)
关键在于本层逻辑的处理
维护一个最大频率maxcount、单数字统计频率count、当前节点root的前一个节点 pre、
class Solution {List<Integer> list = new ArrayList<>();TreeNode pre = null;// 记录前一个节点int maxcount = 0; // 最大频率int count = 0;// 统计频率public int[] findMode(TreeNode root) {dfs(root);return list.stream().mapToInt(Integer::intValue).toArray();}public void dfs(TreeNode root){if(root == null) return;dfs(root.left);if(pre == null) count = 1;// 处理第一个节点else if(root.val == pre.val) count++;// 与前一个节点数值相同else count = 1;// 与前一个节点数值不同pre = root;if(count == maxcount) list.add(root.val);// 如果和最大值相同,放进result中else if(count>maxcount){// 如果计数大于最大值频率maxcount = count;// 更新最大频率list.clear(); // 很关键的一步,不要忘记清空result,之前result里的元素都失效了list.add(root.val);//再把此时的root放进result中}dfs(root.right);}
}