今天参加了个笔试,被问到了这道题,写了半天,面试官都有些不耐烦了,最后也算是写出来了,可是考虑的不够充分,面试后又经过一番优化,总算是感觉相对好一些了,如果有更好的答案欢迎提出!
思路
先说一下思路,我是用Map进行处理的(其实当时有顾虑内存是否能存得下,面试官说能存下),Map的key存放整数值,value存放整数值出现的次数,这样,对数组进行一次遍历之后就可以得到一个存放了整数和对应出现次数的Map了。我本来想的是对value进行排序,但是没想到很好的方法,总想着把value排序之后也没办法获取key,但是事后想了一下,是可以把Map的Entry进行排序的,自定义一个比较器,比较值的内容就是Value,这样只需要找出来排序后最大的n个Entry的key就可以了。
代码
具体的代码实现如下
import java.util.*;
import java.util.stream.Stream;
public class FindNNumber {
public static void main(String[] args) {
Integer[] data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 10, 11, 12, 13, 14, 15,16, 16, 17, 18, 18, 18, 19, 20, 20};
findN(data, 10);
}
public static void findN(Integer[] nums, int n) {
Map<Integer, Integer> map = new HashMap();
//将整数和出现的次数存入Map中
for (int i = 0; i < nums.length; i++) {
int count = map.getOrDefault(nums[i], 0) + 1;
map.put(nums[i], count);
}
//获取Map的Entry,使用流进行排序,自定义Comparator对象,比较的内容是Entry的value值
Stream<Map.Entry<Integer, Integer>> sorted = map.entrySet().stream().sorted(new Comparator<Map.Entry<Integer, Integer>>() {
@Override
public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
//输出前n个元素
final int[] i = {0};
sorted.forEach(o -> {
if (i[0] < n) {
System.out.println(o.getKey());
i[0]++;
}
});
}
}
在这里我用了Stream来处理,当时写的时候也想用来着,但是Stream使用并不熟练,就也没用,就没能很顺利的排序,也算是走了很多弯路吧。
其实不用Stream也是可以的,可以把map的entrySet转换成数组,然后调用Arrays.sort()方法,代码如下:
Map.Entry<Integer, Integer>[] array = new Map.Entry[map.keySet().size()];
map.entrySet().toArray(array);
Arrays.sort(array,(o1,o2)->{
return o2.getValue().compareTo(o1.getValue());
});
for (int i=0;i<n;i++){
System.out.println(array[i].getKey());
}
反思
其实要是有时间还是能够写出来比较完美的程序的,但笔试就是考查有限时间内你的思考和写代码的效率,还是要通过不断的练习提高自己对技术的使用熟练程度啊!