面试题 10.11. 峰与谷-----优先队列
题目表述
在一个整数数组中,“峰”是大于或等于相邻整数的元素,相应地,“谷”是小于或等于相邻整数的元素。例如,在数组{5, 8, 4, 2, 3, 4, 6}中,{8, 6}是峰, {5, 2}是谷。现在给定一个整数数组,将该数组按峰与谷的交替顺序排序。
示例:
示例1:
输入: [5, 3, 1, 2, 3]
输出: [5, 1, 3, 2, 3]
优先队列
只需要以数组的中间为界线,界线左侧的值全部都 < 界线右侧的任意值即可。
class Solution {
public void wiggleSort(int[] nums) {
PriorityQueue<Integer> pq = new PriorityQueue<>();
for(int num : nums){
pq.offer(num);
}
int idx = 0;
if(pq.isEmpty()) return;
for(int i = 0; i < (nums.length - 1) / 2 + 1;i++){
int num = pq.poll();
nums[idx] = num;
idx += 2;
}
idx = 1;
while(!pq.isEmpty()){
nums[idx] = pq.poll();
idx += 2;
}
}
}