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

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

if(list.size()>=k) {
    for(Map.Entry set:list.subList(0, k)) {
        ret.add(set.getKey());
        }
        }
 else {
     for(Map.Entry 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 ret = topKFrequent(nums,k);
			
			for(int x:ret) {
				System.out.print(x+" ");
			}
			System.out.println();
		}
		
		scanner.close();
	}
	
	public static List topKFrequent(int[] nums,int k){
		List ret = new ArrayList();
		if(k>nums.length) {
			return ret;
		}
		
		Map map = new HashMap();
		for(int x:nums) {
			map.put(x, map.getOrDefault(x,0)+1);
		}
		
		List> list = new ArrayList>(map.entrySet());
		Collections.sort(list, new Comparator>() {

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

}

你可能感兴趣的:(Leetcode,前k个高频元素,map排序,Comparetors)