题目描述:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
思路:就是一个排序问题,但只有3种数,因此使用下面的思想:当遍历到数值为num时,把大于等于num的数全在数组上刷上一下,类似于刷油漆,新的油漆覆盖旧的油漆。
类似的,如果有四种值,则用四个num来计算。也是O(n)。
public void sortColors(int[] nums) {
int num0=0,num1=0,num2=0;
for (int i = 0; i < nums.length; i++) {
if (nums[i]==0)
{
nums[num2++]=2;
nums[num1++]=1;
nums[num0++]=0;
}
else if (nums[i]==1)
{
nums[num2++]=2;
nums[num1++]=1;
}
else nums[num2++]=2;
}
}