题目链接
思路:双指针,将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; } }