C/C++教程

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;
    }
}
这篇关于leetcode 颜色分类的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!