1.题目
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
你能想出一个仅使用常数空间的一趟扫描算法吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-colors
2.想法
(1)利用三色问题常用的三指针,a0作为红颜色,代表0;a2作为蓝颜色代表2;扫描一遍数组,扫描每个元素,每次都将当前指针前进一,每遇到0则将a0指针与当前指针互换位置,且a0指针向前移动;每遇到1则不管;每遇到2则将a2指针与当前指针互换位置,且a2指针向后移动,但此时当前指针要减一,因为a2指针还没有扫描过。
3.图解
来源:力扣官方题解
4.自己题解
class Solution {
public void sortColors(int[] nums) {
int a0=0;
int a2=nums.length-1;
for(int i=0;i<nums.length;i++)
{
if(i>a2){break;}
if(nums[i]==0){int temp=nums[a0];nums[a0]=nums[i];nums[i]=temp;a0++;}
else if(nums[i]==2){int temp=nums[a2];nums[a2]=nums[i];nums[i]=temp;a2--;i--;}
}
}
}
5.执行结果