摆动排序 II
给定一个无序的数组 nums
,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]...
的顺序。
示例 1:
输入: nums = [1, 5, 1, 1, 6, 4]
输出: 一个可能的答案是 [1, 4, 1, 5, 1, 6]
示例 2:
输入: nums = [1, 3, 2, 2, 3, 1]
输出: 一个可能的答案是 [2, 3, 1, 3, 1, 2]
说明:
你可以假设所有输入都会得到有效的结果。
进阶:
你能用 O(n) 时间复杂度和 / 或原地 O(1) 额外空间来实现吗?
核心思路:
快速选择 + 3-way-partition + 虚地址
参考: https://leetcode-cn.com/problems/wiggle-sort-ii/solution/yi-bu-yi-bu-jiang-shi-jian-fu-za-du-cong-onlognjia/
https://blog.csdn.net/zjxxyz123/article/details/79900017
class Solution {
public void wiggleSort(int[] nums) {
int k = (nums.length-1)/2;
int left = 0;
int right = nums.length-1;
while(true){
int[] position = partitionThree(nums,left,right);
if(position[0]>k){
right = position[0]-1;
}else{
if(position[1]<k){
left = position[1]+1;
}else{
break;
}
}
}
}
private int[] partitionThree(int[] nums,int left,int right){
int[] result = new int[2];
int temp = nums[getIndex(nums.length,left)];
int i=left-1;
int k= left;
int j= right+1;
while(k<j){
int kk = getIndex(nums.length,k);
if(nums[kk]>temp){
swap(nums,getIndex(nums.length,++i),getIndex(nums.length,k++));
}else if(nums[kk]<temp){
swap(nums,getIndex(nums.length,--j),getIndex(nums.length,k));
}else{
k++;
}
// if(nums[k]>temp){
// swap(nums,++i,k++);
// }else if(nums[k]<temp){
// swap(nums,--j,k);
// }else{
// k++;
// }
}
result[0] = i+1;//中间相同的数的左边界
result[1] = j-1;//中间相同的数的右边界
return result;
}
private int getIndex(int n,int i){
return (2*i+1)%(n|1);
}
private void swap(int[] nums,int l,int r){
int temp = nums[l];
nums[l] = nums[r];
nums[r] = temp;
}
}