Leetcode刷题75(颜色分类)

题目描述：

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

• 一个直观的解决方案是使用计数排序的两趟扫描算法。

首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。

• 你能想出一个仅使用常数空间的一趟扫描算法吗？

 

思路：

 本题为荷兰国旗问题。

1. 通过两次遍历进行排序
2. 设置三个指针，只需要遍历一趟

通过答案：

class Solution {
    public void sortColors(int[] nums) {
        int current=0;        //设置三个指针
        int p1=0;
        int p2=nums.length-1;
        while(current<=p2){      //循环条件
            if(nums[current]==1){     //当前指针的值为1时，当前指针后移
                current++;
            }else if(nums[current]==2){    //当前指针值为2时，与p2指针的值交换，p2前移
                int temp=nums[current];
                nums[current]=nums[p2];
                nums[p2]=temp;
                p2--;
            }else if(nums[current]==0){    //当前指针的值为0时，与p1指针的值交换，p1和当前指针都后移
                int temp=nums[current];
                nums[current]=nums[p1];
                nums[p1]=temp;
                current++;
                p1++;
            }
        }
    }
}