题意:给定一个只包含0,1,2的数组,要求对这数组进行排序
思路:一个简单的思路:记录数组中0,1,2的个数,之后对数组进行重新赋值。该思路需要两次遍历。下面介绍一下只进行一次遍历的方法:
采用双“指针”,设立变量index0和index2记录0和2的位置。即区间都为0,区间
都为2,其中len为数组长度。index0的初值为0,index2的初值为len-1
先做如下设定:输入数组nums,当前遍历位置nums[i]
解题策略如下:
1.nums[i] == 0时,nums[i]与nums[index0]互换,之后index0++,i++
2.nums[i] == 2 && index2 > i时, nums[i]与nums[index2]互换,之后index2--
3.nums[i] == 1时,i++
对于2需要做2点解释:
1).为什么限制index2 > i?
考虑情况:当数组已经排好序,如:[0,0,1,1,2,2]时,若不进行该限制,则会将已排好序的数组打乱。
2).为什么不进行i++?
因为index2所处的位置是没被遍历过的,若是进行了i++,则相当于跳过了一个元素,导致答案错误。
代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int len = nums.size();
if(len <= 1) return;
int index0 = 0, index2 = len - 1;
for(int i=0;i<len;)
{
if(nums[i] == 2 && index2 > i)
{
nums[i] ^= nums[index2];
nums[index2] ^= nums[i];
nums[i] ^= nums[index2];
index2--;
}
else if(nums[i] == 0)
{
nums[i] ^= nums[index0];
nums[index0] ^= nums[i];
nums[i] ^= nums[index0];
index0++;
i++;
}
else i++;
}
}
};