给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库的sort函数的情况下解决这个问题。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
思路:利用双指针,交换、赋值,0 挪到最前面,2 挪到最后面就完事儿了
2 挪完如果换出来的不是 1,那么指针要回退,因为 0 和 2 都是需要再次移动的
public void sortColors(int[] nums) {
int n = nums.length;
if (n < 2){
return;
}
//双指针
int p = 0,q = n-1;
for (int i = 0; i <= q; i++) {
if (nums[i] == 0) {
//把下标i的值换成下标p的值,然后下标p的值为0
nums[i] = nums[p];
nums[p] = 0;
p++;
}
if (nums[i] == 2) {
//把下标i的值换成下标q的值,然后下标q的值为2
nums[i] = nums[q];
nums[q] = 2;
q--;
//例如[1,2,0],先遍历到2,换值0到之前2的位置,0的位置会赋值2;这样会导致输出[1,0,2]
//所以换完值之后,需要判断当前下标i换后是不是1,如果不是则指针回退
if (nums[i] != 1) {
i--;
}
}
}
return;
}