leetcode 324 Wiggle Sort 2

利用中位数的概念，中位数就是将一组数分成2等份（若为奇数，则中位数既不属于左也不属于右，所以是2等份）,其一组数中任何一个元素都大于等于另一组数

那么我们是不是只要一左一右配合着插入，就保证了差值+-+-+-的要求？

 

由于题目输入限制了，必定存在解，所以此处我们不需要担心如果重复中位数太多，出现一左一右相等的情况

算法步骤：

1）找到中位数

　　利用lc215 Kth Largest...，将k=(nums.length+1) / 2，可以在o(n)内求出中位数

2）一左一右插入

　　0 2 4 .. 插大于中位数的

　　1 3 5 .. 插小于中位数的

　　考虑到中位数可能有重复的情况

　　我们从后往前插入小于中位数的值

　　从前往后插入大于中位数的值

　　剩下的空位补上中位数

 

代码中为了省事找中位数的部分用的o(klogn)的方法，详情可以看之前博客

class Solution {
    public void wiggleSort(int[] nums) {
        int median = findKthLargest(nums, (nums.length+1) / 2);
        int left = 1;
        int right = nums.length % 2 == 0 ? nums.length-2 : nums.length-1;
        
        int[] tmp = new int[nums.length];
        for(int i=0; i<nums.length; i++){
            if(nums[i] > median){
                tmp[left] = nums[i];
                left += 2;
            }else if(nums[i] < median){
                tmp[right] = nums[i];
                right -= 2;
            }
        }
        
        while(left < nums.length){
            tmp[left] = median;
            left += 2;
        }
        while(right >= 0){
            tmp[right] = median;
            right -= 2;
        }
        
        System.arraycopy(tmp, 0, nums, 0, nums.length);
    }
    public int findKthLargest(int[] nums, int k) {
        if(nums.length == 0)
            return 0;
        PriorityQueue<Integer> pq = new PriorityQueue<>(Comparator.reverseOrder());
        int res = 0;
        for(int num : nums)
            pq.offer(num);
        
        while(k != 0){
            res = pq.poll();
            k--;
        }
        
        return res;
    }
    
}