LeetCode75——颜色分类（三指针法）

题目描述

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

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

示例：

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

题目分析

题目要得到的结果是把1放在中间，0放在最前面，2放在最后面的一个数组，这一题主要是通过三指针来进行解决的，因为有三个类型的值0，1，2，这三个值可以分别用 P0, P1, P2 这三个指针进行替代。

这样设置三个指针，模仿快排的思路，p0代表下一个可以放置0的位置，p2代表下一个可以放置2的位置，两个指针之间就是应该就是放1的位置。

开始时将p0指向 nums[0] ，p2 指向 nums 数组的最后，p1 用来遍历数组，最开始指向 nums[0] ，并且 p1 只能位于 p0 和 p2 之间，利用 p1 遍历数组过程中有三种情况：
遇到 1 因为此时在 p0 和 p2 之间，所以不管。
遇到 0，将其与 p0 所指向的元素交换，然后 p0 自加1，换回来的肯定是1，所以 p1 也自加1。
遇到 2，和 p2 所指的元素交换，因为换回来的元素不确定是等于多少，所以不能让 p1 自加1，但是 p2 要自减1。

代码

C++ 代码如下：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0=0;//指向放0的位置。
        int p1=0;//用来遍历
        int p2=nums.size()-1;//指向放2的位置
        while(p1<=p2)//所有元素都遍历完了。
        {
            if(nums[p1]==0)swap(nums[p1++],nums[p0++]);//等于0，交换到最前面
            else if(nums[p1]==2)swap(nums[p1],nums[p2--]);//等于2，交换到最后面。
            else p1++;
        }
    }
};