力扣-刷题-324. 摆动排序 II

题目链接

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/wiggle-sort-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

给你一个整数数组 nums，将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]… 的顺序。

你可以假设所有输入数组都可以得到满足题目要求的结果。

示例 1：

输入：nums = [1,5,1,1,6,4]
输出：[1,6,1,5,1,4]
解释：[1,4,1,5,1,6] 同样是符合题目要求的结果，可以被判题程序接受。

示例 2：

输入：nums = [1,3,2,2,3,1]
输出：[2,3,1,3,1,2]

提示：

• 1 <= nums.length <= 5 * 104
• 0 <= nums[i] <= 5000
• 题目数据保证，对于给定的输入 nums ，总能产生满足题目要求的结果

进阶：你能用 O(n) 时间复杂度和 / 或原地 O(1) 额外空间来实现吗？

题目分析

1. 根据题目描述，按规律排序数组
2. 对数组按非递减排序，将数组从中间位置分为两半，依次交叉排序

代码

class Solution {
public:
    void wiggleSort(vector<int>& nums) {
        if (nums.size() == 1) {
            return;
        }
        auto arr = nums;
        std::sort(arr.begin(), arr.end(), compare);
        int mid = (nums.size() + 1) / 2;
        for (int i = 0, j = mid - 1, k = nums.size() - 1; i < nums.size(); i += 2, --j, --k) {
            nums[i] = arr[j];
            if (i + 1 < nums.size()) {
                nums[i + 1] = arr[k];
            }
        }
        return;
    }

private:
    static bool compare(const int a, const int b) {
        return a < b;
    }

};