码迷,mamicode.com
首页 > 其他好文 > 详细

501-二叉搜索树中的众数

时间:2020-01-22 23:51:38      阅读:90      评论:0      收藏:0      [点我收藏+]

标签:tmp   roo   void   hashmap   val   隐式   array   ash   tco   

501-二叉搜索树中的众数

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

  • 结点左子树中所含结点的值小于等于当前结点的值
  • 结点右子树中所含结点的值大于等于当前结点的值
  • 左子树和右子树都是二叉搜索树

例如:
给定 BST [1,null,2,2],

   1
         2
    /
   2

返回[2].

提示:如果众数超过1个,不需考虑输出顺序

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-mode-in-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    public int[] findMode(TreeNode root) {
        // 1.遍历一次,记录所有元素,就可以找到众数
        List<Integer> list = new ArrayList<>();
        list = scan(list, root);

        Integer[] li = list.toArray(new Integer[0]);

        HashMap<Integer, Integer> map = new HashMap<>();
        int max = 0;
        for(Integer i : li) {
            int tmp = map.getOrDefault(i, 0);
            map.put(i, tmp + 1);
            max = Math.max(max, tmp + 1);
        }
        Set<Integer> set = new HashSet<>();
        for(Integer i : map.keySet()) {
            if(map.get(i).equals(max)) {
                set.add(i);
            }
        }

        Integer[] res = set.toArray(new Integer[0]);
        int[] result = new int[res.length];
        for(int i = 0; i < res.length; i++) {
            result[i] = res[i];
        }
        return result;
    }

    private List<Integer> scan(List<Integer> list, TreeNode root) {
        if (root != null) {
            list = scan(list, root.left);
            list = scan(list, root.right);
            list.add(root.val);
        }
        return list;
    }
    int maxTimes = 0;
    int thisTimes = 0;
    List<Integer> res = new LinkedList<>();
    TreeNode pre = null;
    public int[] findMode(TreeNode root) {
        scan(root);
        int[] rr = new int[res.size()];
        for(int i = 0; i < rr.length; i++) {
            rr[i] = res.get(i);
        }
        return rr;
    }

    public void scan(TreeNode root) {
        if(root == null) {
            return;
        }

        scan(root.left);

        if(pre != null && pre.val == root.val) {
            thisTimes++;
        } else {
            thisTimes = 1;
        }
        if(thisTimes == maxTimes) {
            res.add(root.val);
        } else if(thisTimes > maxTimes) {
            maxTimes = thisTimes;
            res.clear();
            res.add(root.val);
        }
        pre = root;

        scan(root.right);
    }

501-二叉搜索树中的众数

标签:tmp   roo   void   hashmap   val   隐式   array   ash   tco   

原文地址:https://www.cnblogs.com/angelica-duhurica/p/12229923.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!