LeetCode 324. Wiggle Sort II(摇摆排序)

时间:2023-02-10 12:59:28

原题网址:https://leetcode.com/problems/wiggle-sort-ii/

Given an unsorted array nums, reorder it such that nums[0] < nums[1] > nums[2] < nums[3]....

Example:
(1) Given nums = [1, 5, 1, 1, 6, 4], one possible answer is [1, 4, 1, 5, 1, 6]
(2) Given nums = [1, 3, 2, 2, 3, 1], one possible answer is [2, 3, 1, 3, 1, 2].

Note:
You may assume all input has valid answer.

Follow Up:
Can you do it in O(n) time and/or in-place with O(1) extra space?

方法一:对数组进行排序,然后将前半段倒序填入奇数下标,将后半段倒序填入偶数下标,其原理是如果题目有解,则对于排好序的数组,间隔超过n/2的两个元素必不相等。时间复杂度O(nlogn),空间复杂度O(n)。

public class Solution {
    public void wiggleSort(int[] nums) {
        int[] sorted = nums.clone();
        Arrays.sort(sorted);
        int mid = (nums.length-1) / 2;
        for(int i=0, j=mid, k=nums.length-1; i<nums.length; i+=2, j--, k--) {
            nums[i] = sorted[j];
            if (i+1<nums.length) nums[i+1] = sorted[k];
        }
    }
}


另一种实现:

public class Solution {
    public void wiggleSort(int[] nums) {
        int[] sort = nums.clone();
        Arrays.sort(sort);
        for(int i=(sort.length-1)/2, j=0; i>=0; i--, j+=2) nums[j]=sort[i];
        for(int i=sort.length-1, j=1; i>(sort.length-1)/2; i--, j+=2) nums[j]=sort[i];
    }
}

此题未搞懂O(n)的算法,留待以后再看。