题目链接
思路:双指针,将0交换到左边,2交换到右边。
有个细节要考虑到,当nums[index]为2的时候,交换到右边,交换回来后的此时的nums[index]还要继续下一轮的判断,不能直接继续后移,当nums[index]为0的时候,交换到左边,此时index可以直接后移,因为此时交换过来的,一定是1,1是放在中间的,所以index可以直接后移。
代码:
class Solution {
public void sortColors(int[] nums) {
int red = 0;
int blue = nums.length - 1;
int index = 0;
while (index <= blue){
if(nums[index] == 0){
swap(nums,index,red++);//此时交换回来的index位置上,一定是1,所以index可以直接后移
index++;
continue;
}
if(nums[index] == 2){
swap(nums,index,blue--);//此时交换回来的可能是1,2,0,所以index不能后移
continue;
}
index++;//此时说明index是1,可以直接后移
}
}
public void swap(int[] nums, int i, int j){
int k = nums[i];
nums[i] = nums[j];
nums[j] = k;
}
}