题目要求
给你一个整数数组 nums,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]... 的顺序。
你可以假设所有输入数组都可以得到满足题目要求的结果。
示例 1:
输入:nums = [1,5,1,1,6,4]
输出:[1,6,1,5,1,4]
解释:[1,4,1,5,1,6] 同样是符合题目要求的结果,可以被判题程序接受。
解法思路
- 对数组进行排序
- 获取数据中间位置: mid = (0+num.length)/2,考虑如果长度是偶数和奇数的情况
如果是偶数,考虑 case:0,1,2,3; 0,0,1,2; 0,1,1,2; 0,1,2,2; 0,0,1,1 这几种情况;实际解法是 nums[0] - nums[mid-1] 倒序 merge 上 nums[mid] - nums[length-1]倒序;上面5种情况结果是:1,3,0,2; 0,2,0,1; 1,2,0,1; 1,2,0,2; 0,1,0,1都满足要求。
如果是奇数,考虑case:0,1,2,3,4; 0,0,1,2,3; 0,1,1,2,3; 0,1,2,2,3; 0,1,2,3,3; 0,0,0,1,2;
0,1,1,1,2;0,1,2,2,2这几种情况;实际解法是nums[0] - nums[mid] 倒序 merge 上 nums[mid+1] - nums[length-1]倒序;2,4,1,3,0; 1,3,0,2,0; 1,3,1,2,0; 2,3,1,2,0; 2,3,1,3,0; 0,2,0,1,0;1,2,1,1,0;2,2,1,2,0; 有删除线的数组实际是没有对应解法的。
代码实现
根据上述解法思路,代码实现如下:
import java.util.Arrays;
class Solution {
public void wiggleSort(int[] nums) {
if (nums.length == 1) {
return;
}
// 做一次排序
Arrays.sort(nums);
boolean even = nums.length % 2 == 0;
int mid = nums.length / 2;
// 将结果写入新的数组
int[] res = new int[nums.length];
if (even) {
// 考虑偶数情况
int index = 0;
int endIndex = nums.length - 1;
int endIndex0 = mid - 1;
while (index < nums.length) {
res[index++] = nums[endIndex0--];
res[index++] = nums[endIndex--];
}
} else {
// 考虑奇数情况
int index = 0;
mid = mid + 1;
int endIndex = nums.length - 1;
int endIndex0 = mid - 1;
while (index < nums.length) {
if (endIndex0 >= 0) {
res[index++] = nums[endIndex0--];
}
if (endIndex >= mid) {
res[index++] = nums[endIndex--];
}
}
}
// 将数据重新设置回去
for (int i = 0; i < nums.length; i++) {
nums[i] = res[i];
}
}
}