leetcode 颜色分类

题目链接
思路:双指针,将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;
    }
}
上一篇:pytorch torch.save()方法保存数据被强制杀死进程,报错:killed 端口号 PYTHONUNBUFFERED=1


下一篇:逆序对计数问题