leetcode 324. Wiggle Sort II

最新推荐文章于 2021-07-12 13:13:05 发布

liqinzhe223

最新推荐文章于 2021-07-12 13:13:05 发布

阅读量251

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/liqinzhe11/article/details/78653211

leetcode 专栏收录该内容

280 篇文章

订阅专栏

本文介绍了一种对数组进行重排序的方法，使得数组中的元素满足特定的交错顺序。通过排序数组并利用中位数特性来实现，或者采用更高效的方法，使用 nth_element 函数快速找到中位数，并在原地完成排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

324. 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?

先对数组进行排序, 然后从左往右奇数索引位置放大于中位数的数, 然后从右往左在偶数索引位置放小于中位数的数, 剩下的位置都放中位数. 其时间复杂度为O(nlog(n)), 空间复杂度为O(n).

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;  
    }
};

网上的大神用的方法！牛的一逼！

class Solution {  
public:  
    void wiggleSort(vector<int>& nums) {  
        nth_element(nums.begin(), nums.begin()+nums.size()/2, nums.end());  
        int len=nums.size(), low=0, high=len-1, mid =nums[len/2], i=0;  
        auto index = [=](int pos){ return (1 + pos * 2) % (len | 1); };  
        while(i <= high)  
        {  
            if(nums[index(i)] > mid) swap(nums[index(i++)], nums[index(low++)]);  
            else if(nums[index(i)]<mid) swap(nums[index(i)],nums[index(high--)]);  
            else i++;  
        }  
    }  
};