题目描述
思路
排序
根据题意要求的新数组元素大小关系可知,将数组从小到大排序后,令
x
=
⌊
n
+
1
2
⌋
x=\lfloor\frac{n+1}{2}\rfloor
x=⌊2n+1⌋,其中n为数组元素个数,必有nums[0]<nums[x],nums[1]<nums[x+1]…,nums[i]<nums[x+i]。
因此可以分类讨论:
- n为偶数时,因为nums[i]<nums[x+i],所以必定满足nums[i]<nums[i+x],nums[i-1]<nums[i+x],因此将nums[i+x]插入到nums[i-1]和nums[i]之间,然后将序列反转即可。
- 当n为奇数时,当满足i>0时,nums[i]≠nums[i+x-1],即nums[i]<nums[i+x-1]。所以有nums[i]<nums[i+x],nums[i+1]<nums[i+x],则可以将nums[i+x]插入到nums[i]和nums[i+1]之间,并反转序列即可。
Python实现
class Solution:
def wiggleSort(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
n = len(nums)
arr = sorted(nums)
x = (n+1) // 2
j, k = x-1, n-1
for i in range(0, n, 2):
nums[i] = arr[j]
if i+1 < n:
nums[i+1] = arr[k]
j -= 1
k -= 1
Java实现
class Solution {
public void wiggleSort(int[] nums) {
int[] arr = nums.clone();
Arrays.sort(arr);
int n = nums.length;
int x = (n+1) / 2;
for (int i = 0, j = x-1, k = n-1; i < n; i += 2, j--, k--) {
nums[i] = arr[j];
if (i+1 < n) {
nums[i+1] = arr[k];
}
}
}
}
C++实现
class Solution {
public:
void wiggleSort(vector<int>& nums) {
int n = nums.size();
vector<int> arr = nums;
sort(arr.begin(), arr.end());
int x = (n + 1) / 2;
for (int i = 0, j = x - 1, k = n - 1; i < n; i += 2, j--, k--) {
nums[i] = arr[j];
if (i + 1 < n) {
nums[i + 1] = arr[k];
}
}
}
};