思考:为什么把add操作中的nums[i] > q.peekLast()换成 大于等于号 就不行了呢?
因为去除操作是按照数值去掉头元素的,假设有一个序列为3,3,在add时,后面的3把前面的3覆盖掉了,如果3是最大元素,在pop时,会把本来应该在窗口里的3,也就是后面的3删除掉,最大值就发生了变化,所以出了问题。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
LinkedList<Integer> q = new LinkedList<>();
int[] res = new int[nums.length - k + 1];
int count = 0;
for (int i = 0; i < k; i ++) {
while (!q.isEmpty() && nums[i] > q.peekLast()) q.pollLast();
q.add(nums[i]);
}
res[count ++] = q.peek();
for (int i = k; i < nums.length; i ++) {
if (!q.isEmpty() && q.peek() == nums[i - k]) q.poll();
while (!q.isEmpty() && nums[i] > q.peekLast()) q.pollLast();
q.add(nums[i]);
res[count ++] = q.peek();
}
return res;
}
}
感觉这题难度在于考察对java集合的使用,想法很简单,先用HashMap把数和对应出现次数存下来,再根据value进行排序,再输出前k个的键值。
class Solution {
public int[] topKFrequent(int[] nums, int k) {
HashMap<Integer, Integer> map = new HashMap<>();
int[] res = new int[k];
int count = 0;
for (int i = 0; i < nums.length; i ++) {
map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
}
ArrayList<Map.Entry<Integer, Integer>> list = new ArrayList<>(map.entrySet());
// 对实体进行排序
list.sort((o1, o2) -> o2.getValue() - o1.getValue());//Lambda表达式
// 输出前k个实体对应的键值
for (Map.Entry<Integer, Integer> a : list) {
res[count ++] = a.getKey();
if (count == k) break;
}
return res;
}
}
后面是改进的算法,使用小顶堆而不是对map进行排序,通过维护一个k大小的小顶堆,该算法时间复杂度为O(nlogk)
class Solution {
public int[] topKFrequent(int[] nums, int k) {
HashMap<Integer, Integer> map = new HashMap<>();
int[] res = new int[k];
for (int i = 0; i < nums.length; i ++) {
map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
}
PriorityQueue<Integer> q = new PriorityQueue<>(new Comparator<Integer>(){
@Override
public int compare(Integer o1, Integer o2) {
return map.get(o1) - map.get(o2);
}
});
for (Integer key : map.keySet()) {
if (q.size() < k) {
q.add(key);
} else if (map.get(key) > map.get(q.peek())){
q.remove();
q.add(key);
}
}
while (!q.isEmpty()) {
res[-- k] = q.poll();
}
return res;
}
}