Leetcode.75 颜色分类

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用库内置的 sort 函数的情况下解决这个问题。

Leetcode.75 颜色分类_第1张图片

 

class Solution {
    public void sortColors(int[] nums) {
        int n0 = 0, n1 = 0;
        //[2,0,2,1,1,0],[1,0,2,2,2]
        for(int i = 0; i < nums.length; i++){
            int num = nums[i];
            nums[i] = 2;
            if(num < 2){
                nums[n1++] = 1;
            }
            if(num < 1){
                nums[n0++] = 0;
            }
        }

    }
}

解析:

我来解释一下。首先,所有数都≤2,那么索性把所有数组置为2,然后遇到所有≤1的,就再全部置为1,,覆盖了错误的2,这时候所有2的位置已经正确,最后所有≤0的全部置为0,也就覆盖了一些错误的1,这时候,0和1的位置都正确。最重要的就是需要两个指针指向下一个1或0的位置

遇到0的情况,先变为1,再变为0,所以现在后面变为1,在前面覆盖0,不会出现0覆盖1的问题

你可能感兴趣的:(算法,leetcode)