324. Wiggle Sort II

今天这道题还是颇有难度，据说是2016年Google的面试题原题，看了大神的代码，简短的三句代码愣是没搞清楚说了啥，好在经过一天的折腾理清了思路。

题目要求是要让数列呈现 第一个数小于第二个数，第二个数大于第三个数，第三个数小于第四个数，以此类推，难点在于要用时间复杂度为O(n)，空间复杂度为O(1)，来解决这个问题。

分析题目我们得到，如果没有时间复杂度和空间复杂度的限制，我们只需要将数字分成元素数目相同的大小两组数（奇数的情况可能差一），然后采用归并排序类似的思想，将大小两组数穿插进入数组中即可，这个思路前半段是可取的，时间复杂度是O(n)，但是后半段是不可取的，因为空间复杂度为O(n)，因此，完整的思路分为两部分：

1、首先，用nth_element()函数，实现中位数的查找，该函数时间复杂度是O(n)，空间复杂度是O(1)。结束之后，需要查找的数在对应的位置上，左边都是比目标数小的数，右边是比目标数大的数。

2、分析题目需要的结果，我们发现，小数都应该分布在偶数索引上，所有的大数都分布在奇数索引上，所以，我们需要借用3-partition的思想，用三个”指针“，分别指向小数，大数和未知数上，当发现检测的数比目标数大，则将目标数和指向大数的指针指向的值相交换，如果发现检测的数比目标数小，则将目标数和指向小数的指针指向的值相交换，目的是将小数都放在0，2，4，6……索引对应的位置上，将大数都放在1，3，5，7……对应的位置上，这部分完成的就是坐标的重新分配，考虑到数列奇数长度和偶数长度的区别，作者巧妙地使用或运算来解决，具体的代码如下：

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