思路一:单指针
1.遍历数组两次,第一次将0元素插在数组开头,第二次将1元素插在0元素后面
2.第一次遍历过程中,设置p0指向0元素待插入的位置,碰上nums[i]==0则与nums[p0]交换,p0++
3.第二次遍历过程与2类似,只不过需要提前确定p1起始位置,而p1起始位置应该是最后一个0元素的后一个元素,也就是p1=p0(end)
代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int p=0;
for(int i=0;i<nums.size();i++)
{
if(nums[i]==0)
{
swap(nums[p],nums[i]);
p++;
}
}
for(int i=0;i<nums.size();i++)
{
if(nums[i]==1)
{
swap(nums[i],nums[p]);
p++;
}
}
}
};
思路二:双指针一
1.从左到右遍历数组,双指针p0和p1分别指向0元素和1元素该插入位置
2.起始p0=p1=0;
3.当nums[i]==1时,交换nums[i]和nums[p1],p1++
4.当nums[i]0时,此时如果p0<p1,则p0指向的值肯定为1,那么交换了nums[i]和nums[p0],会将原来有序的1给交换到nums[i],如果直接p0++就结束的话会对结果造成影响,这时我们只需要按3进行处理即可,也就是交换nums[i]和nums[p1],p1++;此时如果p0p1(也只有上面两种情况),则p1++即可,因为p1所在位置已经固定死了为0。
代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int p0=0;
int p1=0;
for(int i=0;i<nums.size();i++)
{
if(nums[i]==1)
{
swap(nums[p1],nums[i]);
p1++;
}
else if(nums[i]==0)
{
swap(nums[p0],nums[i]);
if(p0<p1)
{
swap(nums[p1],nums[i]);
}
p0++;
p1++;
}
}
}
};
思路三:双指针二
1.遍历整个数组,双指针p0和p2分别指向0元素和2元素该插入位置
2.起始时,p2=最后一个元素下标,p0=0
3.遍历过程中,如果nums[i]==0,交换nums[i]和nums[p0],p0++;
4.如果nums[i]==2,则交换nums[i]和nums[p2],p2++,如果新的nums[i]==0,需要进行处理3;如果新的nums[i]==1,则进入下一次循环;如果新的nums[i]==2,则需要继续交换nums[i]和nums[p2],p2++,直到nums[i]!=2为止。
代码:
class Solution {
public:
void sortColors(vector<int>& nums) {
int p0=0;
int p2=nums.size()-1;
for(int i=0;i<=p2;i++)
{
if(nums[i]==0)
{
swap(nums[i],nums[p0]);
p0++;
}
else if(nums[i]==2)
{
while(nums[i]==2&&i<p2)
{
swap(nums[i],nums[p2]);
p2--;
}
if(i>p2)
{
break;
}
if(nums[i]==0)
{
i--;
}
}
}
}
};