Given an unsorted array nums
, reorder it in-place such that nums[0] <= nums[1] >= nums[2] <= nums[3]...
.
For example, given nums = [3, 5, 2, 1, 6, 4]
, one possible answer is [1, 6, 2, 5, 3, 4]
.
摇摆排序,最容易想到的思路是将整个数组排序,如题例,排序后为1,2,3,4,5,6,然后通过观察很容易可以发现,讲第三个元素3与第二个元素2交换,再将第五个元素5与第四个元素4交换可以符合要求,这样的时间复杂度为O(nlogn),在网上看到一个最优解的复杂度为O(n),解法是这样的:
题目对摇摆排序的定义有两部分:
-
如果i是奇数,
nums[i] >= nums[i - 1]
-
如果i是偶数,
nums[i] <= nums[i - 1]
所以我们只要遍历一遍数组,把不符合的情况交换一下就行了。具体来说,如果nums[i] > nums[i - 1], 则交换以后肯定有nums[i] <= nums[i - 1]。
public void wiggleSort(int[] nums) { for(int i=1;i<nums.length;i++){ if((i%2==1&&nums[i]<nums[i-1])||(i%2==0&&nums[i]>nums[i-1])) swap(nums,i,i-1); } } public void swap(int[] a,int i,int j){ int temp=a[i]; a[i]=a[j]; a[j]=temp; }