Given an unsorted array nums, reorder it such that nums[0] < nums[1] > nums[2] < nums[3]….
Example 1:
Input: nums = [1, 5, 1, 1, 6, 4]
Output: One possible answer is [1, 4, 1, 5, 1, 6].
Example 2:
Input: nums = [1, 3, 2, 2, 3, 1]
Output: One possible answer is [2, 3, 1, 3, 1, 2].
Note:
You may assume all input has valid answer.
分析:
翻译题目的意思,调整列表元素位置,偶数位放较小的,奇数位放较大的。
很容易想到,对列表nums排序,可以用sort,然后对有序列表挑元素重排。
重排时有两个point:如果我们对sortedList,按首尾首尾即1,n,2,n-1,…顺序取,首尾最终在列表中间聚集,这样会有无法AC的情况,如[4,5,5,6]->[4,6,5,5]就Error了。所以要换个想法,不能让两个指针最终在列表中间相遇,要让一个从头(或尾)往中间走,另一个从中间往同方向走。
接着,如果挑元素时从头&中间往后走,而赋给新列表时也按前→后,还是无法AC,如[4,5,5,6]->[4,5,5,6]依然Error。因为排序列表是从小到大,输出列表首位是较小值 (则一定取自左半部分),列表长度为4时,这两个条件导致有序列表右半部分的最小值和有序列表左半部分的最大值依然相邻,我们希望连续两次取的元素在有序列表里是有间隔取的,那么换个方向,赋值新列表按前←后。
(其实若长度>4,按前→后赋值,在有序列表里都是间隔取值的,只有4的时候有特殊情况)
class Solution {
public:
void wiggleSort(vector<int>& nums) {
vector<int> sorted(nums);
sort(sorted.begin(), sorted.end());
int n = sorted.size();
for(int i=nums.size()-1, j=0, k=i/2+1; i>=0; i--){
//j从头往后,k从中间往后,nums从后往前赋值; 若nums从前向后赋值,j和k要从后往前
if(i&1) //i是奇数
nums[i] = sorted[k++];
else
nums[i] = sorted[j++];
}
}
};