出现频率最高的前 K 个元素

题目:有一个不为空且仅包含正整数的数组,找出其中出现频率最高的前 K 个数,时间复杂度必须在 O(n log n) 以内。

输入:一行数据包括两部分,一个正整数数组(数字间 ',' 分隔)和一个正整数 K (1 ≤ K ≤ 数组长度),数组和 K 之间有一个空                格。

输出:输出包含前 K 个出现频率最高的数(出现频率相同时,较小的数在前),用 ', ' 分隔,保证升序排列。

输入样例:1,1,1,2,2,3 2

输出样例:1,2

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;

public class frequency {
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		System.out.println(solution(in.nextLine()));
		in.close();
	}
	
	private static String solution(String line) {
		String[] array=line.split(" ");
		String[] numbers=array[0].split(",");
		int k=Integer.parseInt(array[1]);
		List<Integer> num1=new ArrayList<Integer>(); 
		for(int i=0;i<numbers.length;i++){
			num1.add(Integer.parseInt(numbers[i]));
		}
		//提取不重复的数字到list中
		List<Integer> list=new ArrayList<Integer>(); 
		list.add(num1.get(0));
		int n=0;
		int s=0;
		for(int j=1;j<num1.size();j++){
			s=0;
			if(list.get(n)!=num1.get(j)){
				for(int m:list){
					if(num1.get(j)==m)
						s=1;
				}
				if(s==0){
					list.add(num1.get(j));
					n++;
				}
			}
		}
		Collections.sort(list);
		
		//查询每个数字出现的次数存储在map的value中,以list中的数当key
		Map<String,String> map=new HashMap<String,String>();
		for(int p:list){
			int count=0;
			for(int i=0;i<num1.size();i++){
				if(p==num1.get(i))
					count+=1;
			}
			map.put(p+"",count+"");
		}
		List<Map.Entry<String,String>> newmap= mm(map);
		String res="";
		 for(Map.Entry<String,String> mapping:newmap){ 
			 if(k==0)
				 break;
			 res+=mapping.getKey()+",";
			 --k;
//			 System.out.println(mapping.getKey()+":"+mapping.getValue()); 
		 } 
		 res = res.substring(0,res.length()-1);
		return res;
	}
	
	//按照map的value值排序
	public static List<Map.Entry<String,String>> mm(Map<String, String> map){
		List<Map.Entry<String,String>> list = new ArrayList<Map.Entry<String,String>>(map.entrySet());
        //然后通过比较器来实现排序
        Collections.sort(list,new Comparator<Map.Entry<String,String>>() {
            //升序排序
            public int compare(Map.Entry<String, String> o1,
            		Map.Entry<String, String> o2) {
                return o2.getValue().compareTo(o1.getValue());
            }
        });
        return list;
	}

}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值