1. 程式人生 > >[Leetcode] 第324題 擺動排序II

[Leetcode] 第324題 擺動排序II

一、題目描述

給定一個無序的陣列 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) 額外空間來實現嗎?

二、題目解析

1)首先對無序陣列進行排序,使用額外的空間tmp
2)排序之後,進行取數,從中間分成左右兩部分,分別從兩部分的末尾開始取
3)這樣的時間複雜度是n*logn,空間複雜度是n

三、程式碼實現

 1 class Solution {
 2 public:
 3     void wiggleSort(vector<int>& nums) {
 4         int n = nums.size();
 5         if (n < 2)return;
 6         vector<int>tmp = nums;
 7         sort(tmp.begin(), tmp.end());
8 int k = (n + 1) / 2, t = n;//如果是奇數,那麼中間的數應該是屬於前半部分的 9 for (int i = 0; i < n; ++i) { 10 nums[i] = i % 2 ? tmp[--t] : tmp[--k]; 11 } 12 } 13 };