原题网址:https://leetcode.com/problems/wiggle-sort-ii/
Given an unsorted array nums
, reorder it such that nums[0] < nums[1] > nums[2] < nums[3]...
.
Example:
(1) Given nums = [1, 5, 1, 1, 6, 4]
, one possible answer is [1, 4, 1, 5, 1, 6]
.
(2) Given nums = [1, 3, 2, 2, 3, 1]
, one possible answer is [2, 3, 1, 3, 1, 2]
.
Note:
You may assume all input has valid answer.
Follow Up:
Can you do it in O(n) time and/or in-place with O(1) extra space?
方法一:对数组进行排序,然后将前半段倒序填入奇数下标,将后半段倒序填入偶数下标,其原理是如果题目有解,则对于排好序的数组,间隔超过n/2的两个元素必不相等。时间复杂度O(nlogn),空间复杂度O(n)。
public class Solution { public void wiggleSort(int[] nums) { int[] sorted = nums.clone(); Arrays.sort(sorted); int mid = (nums.length-1) / 2; for(int i=0, j=mid, k=nums.length-1; i<nums.length; i+=2, j--, k--) { nums[i] = sorted[j]; if (i+1<nums.length) nums[i+1] = sorted[k]; } } }
另一种实现:
public class Solution { public void wiggleSort(int[] nums) { int[] sort = nums.clone(); Arrays.sort(sort); for(int i=(sort.length-1)/2, j=0; i>=0; i--, j+=2) nums[j]=sort[i]; for(int i=sort.length-1, j=1; i>(sort.length-1)/2; i--, j+=2) nums[j]=sort[i]; } }
此题未搞懂O(n)的算法,留待以后再看。