Leetcode 324. 摆动排序 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) 额外空间来实现吗？

用一个辅助数组，排序后。
nums数组的奇数下标存放小的数，偶数下标存放大的数。
O（nlogn）方法。

class Solution {
public:
    void wiggleSort(vector<int>& nums) 
    {
        vector<int>tmp;
        tmp=nums;
        int len=nums.size();
        sort(tmp.begin(),tmp.end());
        int k=(len+1)/2,j=len;
        for(int i=0;i<tmp.size();i++)
        {
            if(i%2==0)
            {
                nums[i]=tmp[--k];
            }
            else
            {
                nums[i]=tmp[--j];
            }
        }
    }
};

 

下面给出O(n)的方法，挺难的，可以自己钻研一下。

class Solution {
public:
    void wiggleSort(vector<int>& nums) {
        #define A(i) nums[(1 + 2 * i) % (n | 1)]
        int n = nums.size(), i = 0, j = 0, k = n - 1;
        auto midptr = nums.begin() + n / 2;
        nth_element(nums.begin(), midptr, nums.end());
        int mid = *midptr;
        while (j <= k) {
            if (A(j) > mid) swap(A(i++), A(j++));
            else if (A(j) < mid) swap(A(j), A(k--));
            else ++j;
        }
    }
};