280. 摆动排序

280. 摆动排序

给你一个无序的数组 nums, 将该数字 原地 重排后使得 nums[0] <= nums[1] >= nums[2] <= nums[3]...。

示例:

输入: nums = [3,5,2,1,6,4]
输出: 一个可能的解答是 [3,5,1,6,2,4]

解法一：排序

思路

排序后，从第二元素开始每对交换

[3,5,2,1,6,4] 排序 [1,2,3,4,5,6]
[1, 2,3, 4,5, 6] 第二个元素开始按对交换 [1, 3,2, 5,4, 6] 满足

复杂度

时间复杂度：$O(NlogN)$

空间复杂度：$O(1)$

代码

class Solution {
    public void wiggleSort(int[] nums) {
        Arrays.sort(nums);
        for (int i = 1; i < nums.length; i++) {
            swap(nums, i - 1, i);
        }
    }
    void swap(int[] a, int i, int j) {
        int t = a[i];
        a[i] = a[j];
        a[j] = t;
    }
}

解法二：一次遍历

思路

就是按照定义模拟，这个方法比上面那个好想一点。

复杂度

时间复杂度：$O(N)$

空间复杂度：$O(1)$

代码

class Solution {
    public void wiggleSort(int[] nums) {
        boolean flag = true;
        for (int i = 1; i < nums.length; i++) {
            if (flag) {
                if (nums[i] < nums[i - 1]) swap(nums, i, i - 1);
            } else {
                if (nums[i] > nums[i - 1]) swap(nums, i, i - 1);
            }
            flag = !flag;
        }
    }
    void swap(int[] a, int i, int j) {
        int t = a[i];
        a[i] = a[j];
        a[j] = t;
    }
}