其他分享
首页 > 其他分享> > leetcode 颜色分类

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;
    }
}

标签:index,颜色,nums,int,分类,交换,后移,swap,leetcode
来源: https://blog.csdn.net/weixin_42496727/article/details/120573617