题意:Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.
Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.
难度:Medium
解题思路:看到题目首先想到的是排序,但是考虑到数组中只有三种类型的元素,同一种类型的元素不加区分。 可以用三个变量a,b,c来表示各类元素的数目,遍历一次数组就可以统计出来。 然后再遍历一次数组,下标小于a的就是第一种元素的位置,下标在a-a+b之间的就是第二种元素,除此之外是第三种元素。这种算法遍历两次可以得到最后的结果,时间复杂度为O(n)
class Solution {
public:
void sortColors(vector<int>& nums) {
int a=0,b=0;
int x=0,y=0,z=0;
for(int i = 0;i<nums.size();i++)
{
if(nums[i]==0)
a++;
if(nums[i]==1)
b++;
}
for(int i = 0;i<nums.size();i++)
{
if(i<a)
{
nums[i]=0;
}
else if(i<a+b)
{
nums[i]=1;
}
else nums[i]=2;
}
}
};