Leetcode---前K个高频元素--O(n)复杂度

前K个高频元素

题目链接:前K个高频元素

思路:

解法一:

  • 首先对整个数组排序,复杂度O(nlgn)
  • 再对每个数字出现的频度排序
  • 最后找出前K个数字

解法二:

  • 将数组遍历存入map集合中,value值存放出现次数
  • 构造一个长度为nums.length+1的lists数组
  • 将频度作为下标添加至lists[i]集合中
  • 最后按标从大到小遍历,获取集合中的值,添加至结果集,加入k个即可
	public List<Integer> topKFrequent(int[] nums, int k) {
		
		Map<Integer,Integer> map = new HashMap<Integer,Integer>();
		for(int num:nums) {
			map.put(num,map.getOrDefault(num, 0)+1);
		}
		//这里加1的原因是不必单独处理下标
		List[] lists = new ArrayList[nums.length+1];
		for(int key:map.keySet()) {
			//取出key值对应的元素个数
			int temp = map.get(key);
			if(lists[temp]==null) {
				lists[temp] = new ArrayList<Integer>();
			}
			//这里就是用list存储的原因,因为同一个value值可能对应着好几个key值
			lists[temp].add(key);
		}
		List<Integer> result = new ArrayList<Integer>();
		for(int i=lists.length-1;result.size()<k&&i>=0;--i) {
			if(lists[i]!=null) {
				result.addAll(lists[i]);
			}
		}
		return result;
    }

猜你喜欢

转载自blog.csdn.net/tiaochewang219/article/details/84886122
今日推荐