题目地址:
https://leetcode.com/problems/wiggle-sort/
给定一个数组 A A A,重排之使得 A [ 0 ] ≤ A [ 1 ] ≥ A [ 2 ] ≤ A [ 3 ] ≥ A [ 4 ] . . . A[0]\le A[1] \ge A[2]\le A[3]\ge A[4]... A[0]≤A[1]≥A[2]≤A[3]≥A[4]...。
从左到右遍历数组,如果发现大小关系不符合就直接swap两个数即可。代码如下:
public class Solution {
public void wiggleSort(int[] nums) {
if (nums == null || nums.length <= 1) {
return;
}
for (int i = 0; i < nums.length - 1; i++) {
if (i % 2 == 0) {
if (nums[i] > nums[i + 1]) {
swap(nums, i, i + 1);
}
} else {
if (nums[i] < nums[i + 1]) {
swap(nums, i, i + 1);
}
}
}
}
private void swap(int[] A, int i, int j) {
int tmp = A[i];
A[i] = A[j];
A[j] = tmp;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。
算法正确性证明:
发现不符合的情况,只能是出现了单调的三个数(可以想象
A
[
0
]
A[0]
A[0]前面有一个正无穷大,这样一开头不符合也可以算是出现了单调的三个数),无论是单调增还是单调减,我们都可以发现,只需要调换后两个数就行了。