leetcode笔记:Wiggle Sort

来源:互联网 发布:网页美工视频教程 编辑:程序博客网 时间:2024/05/23 23:48

一. 题目描述

Given an unsorted array nums, reorder it in-place such that nums[0] <= nums[1] >= nums[2] <= nums[3]

For example, given nums = [3, 5, 2, 1, 6, 4], one possible answer is [1, 6, 2, 5, 3, 4].

二. 题目分析

题目给出一个未排序的数组,调整元素的大小使其满足nums[0] <= nums[1] >= nums[2] <= nums[3],要求in-place。

第一种思路是先排序后调整。根据题目的描述,数组的组合方式有多种。因此可以先将数组进行排序,这时候从第3个元素开始,将第奇数个元素与前面一个偶数位置元素进行交换,如将第3个元素和第2个元素交换、将第5个元素和第4个元素交换,以此类推。该方法的时间复杂度为:O(NlogN),空间复杂度:O(1)

第二中思路是greedy,使用这种方法只需遍历一次数组,时间复杂度为:O(N)。同样满足题目的in-place要求。我们知道,在题目的描述中可以总结出以下规律:

当下标i是奇数时,nums[i] >= nums[i - 1]
当下标i是偶数时,nums[i] <= nums[i - 1]

根据以上规律,只需遍历一遍数组,把不符合的条件的元素交换一下即可。

三. 示例代码

// 先排序后调整class Solution {public:    void wiggleSort(vector<int>& nums) {        if(nums.size() < 2) return;          sort(nums.begin(),nums.end());        // 将数组中一对一对交换        for(int i = 2; i < nums.size(); i += 2){            int tmp = nums[i-1];            nums[i-1] = nums[i];            nums[i] = tmp;        }    }};
// 贪心法class Solution {public:    void wiggleSort(vector<int>& nums) {        if(nums.size() < 2) return;          for (int i = 1; i < nums.size(); ++i) {            if((i % 2 == 1 && nums[i] < nums[i-1]) || (i % 2 == 0 && nums[i] > nums[i-1]))                 swap(nums[i], nums[i-1]);        }    }};

四. 小结

后续有题目:Wiggle Sort II。

3 0
原创粉丝点击