Wiggle Sort

Wiggle Sort

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].

排序法

复杂度

时间 O(NlogN) 空间 O(1)

思路

根据题目的定义，摇摆排序的方法将会很多种。我们可以先将数组排序，这时候从第3个元素开始，将第3个元素和第2个元素交换。然后再从第5个元素开始，将第5个元素和第4个元素交换，以此类推。就能满足题目要求。

代码

public class Solution {
    public void wiggleSort(int[] nums) {
        // 先将数组排序
        Arrays.sort(nums);
        // 将数组中一对一对交换
        for(int i = 2; i < nums.length; i+=2){
            int tmp = nums[i-1];
            nums[i-1] = nums[i];
            nums[i] = tmp;
        }
    }
}

交换法

复杂度

时间 O(N) 空间 O(1)

思路

题目对摇摆排序的定义有两部分：

1. 如果i是奇数，nums[i] >= nums[i - 1]

2. 如果i是偶数，nums[i] <= nums[i - 1]

所以我们只要遍历一遍数组，把不符合的情况交换一下就行了。具体来说，如果nums[i] > nums[i - 1]， 则交换以后肯定有nums[i] <= nums[i - 1]。

代码

public class Solution {
    public void wiggleSort(int[] nums) {
        for(int i = 1; i < nums.length; i++){
            // 需要交换的情况：奇数时nums[i] < nums[i - 1]或偶数时nums[i] > nums[i - 1]
            if((i % 2 == 1 && nums[i] < nums[i-1]) || (i % 2 == 0 && nums[i] > nums[i-1])){
                int tmp = nums[i-1];
                nums[i-1] = nums[i];
                nums[i] = tmp;
            }
        }
    }
}

class Solution {
public:
    void wiggleSort(vector<int>& nums) {
        int len = nums.size();
        int mid = len/2;
        int index = quickSort(nums,0,len-1);
        while(index != mid)
        {
            if(index > mid)
                index = quickSort(nums,0,index-1);
            else
                index = quickSort(nums,index+1,len-1);
        }
        int j = 1;
        for(int i=mid; i<len; i+=2)
        {
            swap(nums[j],nums[i]);
            j += 2;
        }
    }
    int quickSort(vector<int>& nums,int low, int high)
    {
        int pivot = nums[low];
        while(low < high)
        {
            while((low < high) &&(nums[high] >= pivot))
                --high;
            nums[low] = nums[high];
            while((low < high) &&(nums[low] <= pivot))
                ++low;
            nums[high] = nums[low];
        }
        nums[low] = pivot;
        return low;
    }
};