给定一个包含红色、白色和蓝色、共
n
个元素的数组nums
,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。我们使用整数
0
、1
和2
分别表示红色、白色和蓝色。必须在不使用库内置的 sort 函数的情况下解决这个问题。
class Solution {
public void sortColors(int[] nums) {
int n0 = 0, n1 = 0;
//[2,0,2,1,1,0],[1,0,2,2,2]
for(int i = 0; i < nums.length; i++){
int num = nums[i];
nums[i] = 2;
if(num < 2){
nums[n1++] = 1;
}
if(num < 1){
nums[n0++] = 0;
}
}
}
}
解析:
我来解释一下。首先,所有数都≤2,那么索性把所有数组置为2,然后遇到所有≤1的,就再全部置为1,,覆盖了错误的2,这时候所有2的位置已经正确,最后所有≤0的全部置为0,也就覆盖了一些错误的1,这时候,0和1的位置都正确。最重要的就是需要两个指针指向下一个1或0的位置
遇到0的情况,先变为1,再变为0,所以现在后面变为1,在前面覆盖0,不会出现0覆盖1的问题