给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
思路:
思路1:快排
思路2:单指针
先遍历一遍,把为0的元素交换到数组的头部。然后第二次遍历把为1的元素交换到0元素的后面。
交换完0和1以后,2也到了数组的最后。
和移动零的思路一样都是交换,这个交换了两次。
class Solution {
public void sortColors(int[] nums) {
int p =0;
int n = nums.length;
for(int i=0;i<n;i++)
{
if(nums[i]==0)//如果遍历到0元素,则交换到开头
{
nums[i] = nums[p];
nums[p] = 0;
p++;
}
}
for(int i=p;i<n;i++)
{
if(nums[i]==1)
{
nums[i] = nums[p];
nums[p] = 1;
p++;
}
}
}
}