leetcode-前k个高频元素

题目：给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

说明：
你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

题解：
将数组值作为key，出现的频次作为value,遍历数组，构建HashMap，然后根据map的value进行排序，之后遍历排序好的map，将前k个出现频次最高的Key返回.

1.构建map，map.getOrDefault(x,0)会对首次key的value值进行默认值赋值

Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for(int x:nums) {
    map.put(x, map.getOrDefault(x,0)+1);
}

2.对map根据value值进行排序，Collections.sort第一个参数需要传入list类型，所以将map中的键值对封装为list传入，o2.getValue()-o1.getValue()表示降序排序

List<Map.Entry<Integer, Integer>> list = new ArrayList<Map.Entry<Integer,Integer>>(map.entrySet());
Collections.sort(list, new Comparator<Entry<Integer, Integer>>() {
    @Override
    public int compare(Entry<Integer, Integer> o1, Entry<Integer, Integer> o2) {
        // TODO 自动生成的方法存根
            return o2.getValue()-o1.getValue();
      }
 });

3.从排好序的map中返回前k个出现频次最高的key
注意：因为sort方法是对List中的键值对进行排序的，所以我们需要遍历List才可以保证我们访问的排序顺序。

if(list.size()>=k) {
    for(Map.Entry<Integer, Integer> set:list.subList(0, k)) {
        ret.add(set.getKey());
        }
        }
 else {
     for(Map.Entry<Integer, Integer> set:list) {
         ret.add(set.getKey());
         }
  	}

整体算法：

package com.ljq.sort;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;

/** 
* @author 作者 Your-Name: 
* @version 创建时间：2019年8月27日 下午3:21:23 
* 类说明 
*/
public class topK {

	public static void main(String[] args) {
		// TODO 自动生成的方法存根
		Scanner scanner = new Scanner(System.in);
		
		while(scanner.hasNext()) {
			String[] strings = scanner.nextLine().split(" ");
			int len = strings.length;
			
			int k = Integer.parseInt(strings[0]);
			int[] nums = new int[len-1];
			for(int i=0;i<len-1;i++) {
				nums[i] = Integer.parseInt(strings[i+1]); 
			}
			
			List<Integer> ret = topKFrequent(nums,k);
			
			for(int x:ret) {
				System.out.print(x+" ");
			}
			System.out.println();
		}
		
		scanner.close();
	}
	
	public static List<Integer> topKFrequent(int[] nums,int k){
		List<Integer> ret = new ArrayList<Integer>();
		if(k>nums.length) {
			return ret;
		}
		
		Map<Integer, Integer> map = new HashMap<Integer, Integer>();
		for(int x:nums) {
			map.put(x, map.getOrDefault(x,0)+1);
		}
		
		List<Map.Entry<Integer, Integer>> list = new ArrayList<Map.Entry<Integer,Integer>>(map.entrySet());
		Collections.sort(list, new Comparator<Entry<Integer, Integer>>() {

			@Override
			public int compare(Entry<Integer, Integer> o1, Entry<Integer, Integer> o2) {
				// TODO 自动生成的方法存根
				return o2.getValue()-o1.getValue();
			}
		});
		
		if(list.size()>=k) {
			for(Map.Entry<Integer, Integer> set:list.subList(0, k)) {
				ret.add(set.getKey());
			}
		}else {
			for(Map.Entry<Integer, Integer> set:list) {
				ret.add(set.getKey());
			}
		}
		
		return ret;
	}

}