题目要求在一次遍历内完成三种颜色的分类,那么可以考虑对分类完成后位于数组两端的颜色进行分类,第三种颜色自然会落在数组中间。
使用双指针l和r,分别指向数组开头和末尾,然后从数组开头遍历数组,当数值为0时,与l指针对应的元素交换,并且同时+1,向后移位;当数值为2时,与r指针对应的元素交换。
这里要特别注意,交换完成后,r指针-1,向前移位,而由于交换后,当前元素的数值未知,因此此时遍历指针不能+1,而是继续判断该元素的大小。
**class Solution {
public void sortColors(int[] nums) {
int l = 0;
int r = nums.length-1;
int temp;
int i=0;
while(i<=r){
if(nums[i] == 0){
temp = nums[l];
nums[l++] = nums[i];
nums[i++] = temp;
}else if(nums[i] == 2){
temp = nums[r];
nums[r--] = nums[i];
nums[i] = temp;
}else{
i++;
}
}
}
}**