题意
给定一个数组,只包含0,1,2,将0,1,2按照顺序排列。要求:
1. 只能开常数空间
2. 对数组只扫描一遍
思路
算法1
算法1是最容易想到的,我们扫描一遍数组统计一下每个数字的数目,然后再扫描一遍填进去即可。
算法2
我们可以这样想,只需要确定每个数字的开头位置即可。
首先,0的开头位置一定是0,那么,我们只需要确定1和2的开头位置即可。于是,我们设置两个指针ind1和ind2,分别代表1的开头位置,和2的开头位置的前一个数。
然后,设置一个指针i开始对数组进行扫描,并且,我们可以很容易分析出
ind1≤i≤ind2
:
- 若 a[i]==0 :那么,将a[i]和a[ind1]交换位置,ind1++,i++。
- 若 a[i]==1 :那么,ind1不用右挪,只需要i++即可。
- 若 a[i]==2 :那么,我们需要将a[i]交换到后面去,因为ind2指向的是2前面的一个数,因此,我们交换a[i]和a[ind2]后,肯定ind2–,因为2的开头位置会减小,但是i不能++,因为我们指向的是2前面的一个数字,可能为1或0,因此还需要继续一轮判断。
其算法的思想类似于快排的partition。
算法3
非常巧妙的一个算法,其思想在于平移。
当我们遇到一个0的时候,所有数的起始点都要后挪一位,因此:
a[++ind2] = 2; a[++ind1] = 1; a[++ind0] = 0;
当我们遇到1的时候,只有1和2的起始点会挪,因此:
a[++ind2] = 2; a[++ind1] = 1;
当我们遇到2的时候,只有2会挪,因此:
a[++ind2] = 2;
算法4
自己的算法是设置3个指针:i, j, newi。其中,i向右扫描,j向左扫描:
当a[i] == 0时:a[newi] = 0;
当a[i] == 1时,直接i++;
当a[i] == 2时,这时候,我们要把2放后面,j指向的是2的开头位置的前一个。然后,我们再判断a[j]位置的数:
1. 若a[j] == 2:j前挪到第一个不是2的位置
2. 若a[j] == 1:直接将a[j]替换为2
3. 若a[j] == 0:将a[j]替换为2,并且a[newi]替换为0
代码
//algorithm 2, based on two index
class Solution {
public:
void sortColors(vector<int>& nums) {
int i = 0, ind1 = 0, ind2 = nums.size() - 1;
while (i <= ind2) {
if (nums[i] == 0) swap(nums[i++], nums[ind1++]);
else if (nums[i] == 1) i++;
else swap(nums[i], nums[ind2--]);
}
}
};
//algorithm3, based on shift
class Solution {
public:
void sortColors(vector<int>& nums) {
int ind1 = -1, ind2 = -1, ind3 = -1;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == 0) {
nums[++ind3] = 2;
nums[++ind2] = 1;
nums[++ind1] = 0;
}
else if (nums[i] == 1) {
nums[++ind3] = 2;
nums[++ind2] = 1;
} else {
nums[++ind3] = 2;
}
}
}
};
//algorithm3, based on index
class Solution {
public:
void sortColors(vector<int>& a) {
int i = 0, j = a.size() - 1, newi = 0;
while (i <= j) {
if (a[i] == 0) {
a[newi++] = 0;
} else if (a[i] == 2) {
while (a[j] == 2 && j > i) j--;
if (a[j] == 0) {
a[newi++] = 0;
a[j--] = 2;
} else {
a[j--] = 2;
}
}
i++; //i always move right
}
while (newi <= j) a[newi++] = 1;
}
};