来源:力扣(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]
提示:
进阶:你能用 O(n) 时间复杂度和 / 或原地 O(1) 额外空间来实现吗?
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; } };