【编程题】调整红白蓝三色

/*******************
思想：
 1.设置两个指针  start指向第一个元素  end指向最后一个元素
 2.start找到非0元素，end找到0元素
 3.交换两者，回到2
 4.此时所有0元素在开头，继续遍历交换 1 2 元素
 时间复杂度：
   O(2n)
**********************/
void sortColors(vector<int>& nums) {
int start = 0;
int end = nums.size() - 1;


//将所有的0交换到前面
while (start < end)
{
while (start<end &&nums[start] == 0)
++start;
while (start<end && nums[end] != 0)
--end;
if (start < end)
{
nums[start] = nums[start] + nums[end];
nums[end] = nums[start] - nums[end];
nums[start] = nums[start] - nums[end];

}


}


//将1 2 交换
end = nums.size()-1;
while (start < end)
{
while (start<end && nums[end] == 2)
--end;
while (start<end &&nums[start] == 1)
++start;
if (start < end)
{
nums[start] = nums[start] + nums[end];
nums[end] = nums[start] - nums[end];
nums[start] = nums[start] - nums[end];
}
}
}