题目分析:给定一个数组,数组中的元素只可能为0,1,2,其中0代表红色,1代表白色,2代表蓝色。请按照红、白、蓝的顺序排好。不能使用排序算法实现。
解题思路:
一遍扫描遍历数组,需要三个指针进行记录,详细如下:
1)从开始起第一个不确定为0的位置;
2)遍历指针所处的当前位置;
3)从末尾起第一个不确定是否为2的位置;
如果当前指针指向的值为0,则与前面指针指向的值进行交换,并向前移动两个指针的位置;
如果当前指针指向的值为1,则将当前指针向前移;
如果当前指针指向的值为2,则与后面指针指向的值进行交换,并向前移动指向后面的指针。
class Solution
{
public:
// 交换两个数函数
void swap(int &a, int &b)
{
int temp = a;
a = b;
b = temp;
}
//直观理解
void sortColors1(vector<int> &nums)
{
int len = nums.size();
if (len <= 1)
return ;
int i = 0;
int j = len - 1;
int cur = i;
while (cur <= j)
{
// 当前指针指向值为0时,当前指针与指向第一个不确定为0的指针交换,同时向前移动连个指针
if (nums[cur] == 0)
{
swap(nums[i++], nums[cur++]);
}
// 当前指针指向值为2时,当前指针与指向第一个不确定为2的指针交换,同时向前移动后面的指针
else if (nums[cur] == 2)
{
if (cur < j)
{
swap(nums[j--], nums[cur]);
}
else
return ;
}
// 当前指针指向值为0时,向前移动当前真真
else
{
cur++;
}
}
}
// 与上面的原理一样,只是便于理解。
void sortColors(vector<int> &nums)
{
int zero_end = -1;
int two_begin = nums.size();
int i = 0;
while (i < two_begin)
{
if (nums[i] == 0 && i != ++zero_end)
{
swap(nums[i], nums[zero_end]);
}
else if (nums[i] == 2 && i != --two_begin)
{
swap(nums[i], nums[two_begin]);
}
else
i++;
}
}
};