【LeetCode】324. 摆动排序 II 结题报告 (C++)

原题地址:https://leetcode-cn.com/problems/wiggle-sort-ii/submissions/

题目描述:

给定一个无序的数组 nums,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]... 的顺序。

示例 1:

输入: nums = [1, 5, 1, 1, 6, 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) 额外空间来实现吗?

解题方案:

排序问题,先使用sort()函数进行排序,再对数据重新定位。

代码:

class Solution {
public:
    void wiggleSort(vector<int>& nums) {
        if (nums.size() <= 1) 
            return;
        sort(nums.begin(), nums.end());
        int len = nums.size(), k = 1, high = (len % 2) ? len - 1 : len - 2, mid = nums[len / 2];
        vector<int> ans(len, mid);
        
        for (int i = len - 1; i >= 0 && nums[i] > mid; --i, k += 2) 
            ans[k] = nums[i];
        for (int i = 0; i < len && nums[i] < mid; ++i, high -= 2) 
            ans[high] = nums[i];
        nums = ans;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_32805671/article/details/86941749