Leetcode | 0075.单词分类

题目

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

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

示例

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

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

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

解决方法

对数组进行遍历，遍历的同时对数组中的元素进行交换。交换的目的：将数组中所有的0交换到数组的头部，将数组中所有的1交换到头部的0之后。此时，所有的2都出现在数组的尾部.

使用两个指针，第一个指针p1表示“头部”的范围，指向“头部”最后一个元素的下一个元素。第二个指针p2表示元素为1的范围，指向元素为1的区域的下一个元素。

 从左向右遍历整个数组，如果遇到了1，则将其与nums[p1]进行交换，并将p1向后移动一个位置（可以理解为给元素为1的区间添加位置）。

如果在遍历过程中遇到了0，则将其与nums[p0]进行交换，并将p0向后移动一个位置。需要注意的是，当p0<p1时，我们已经将一些1连续地放在前面，此时一定会把一个1交换出去，导致答案错误。

因此，当p0<p1时，需要再将nums[p1]和nums[i]进行交换。在进行第一次交换后，nums[i]的值为1，需要将这个1放到属于1的区域。

最后，同时将p0和p1向后移动一个位置。

代码实现

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p1 = 0; 
        int p2 = 0; 
        for(int i = 0;i < nums.size();i ++){
            if(nums[i] == 1){
                swap(nums[p2],nums[i]);
                p2 ++;
            }else  if(nums[i] == 0){
                swap(nums[p1],nums[i]);
                if(p1 < p2){
                    swap(nums[p2],nums[i]);
                }
            p2 ++;
            p1 ++;
            }
        }
        return;
    }
};