leetcode刷题之75.颜色分类

题目描述

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

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

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

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

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

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

解题思路

简单：排序

（0.0不赘述了）
这里的数据只有0、1、2三种，可以使用更加特殊的方法做。

简单：2次遍历

第一次遍历：用3个int，统计3种颜色的个数，假设有红色x个、白色y个和蓝色z个

第二次遍历：赋值。
0 ~ (x-1) 红色
x ~ (x+y-1)白色
x+y ~ n-1 蓝色

//第1次遍历：用3个int，统计3种颜色的个数
int cnt[3]={0};
for(int i=0;i<nums.size();i++){
	cnt[nums[i]]++;
}

//第2次遍历
for(int i=0;i<nums.size();i++){
	if(i<x) nums[i]=0;
	else if(i<x+y) nums[i]=1;
	else nums[i]=2;
}

进阶：1次遍历

0.一开始思路是用双指针，一个从前往后负责识别0，一个从后往前负责识别2。对比前大于后,则交换

但这样是错误的！原因：中间1的区间中指针没办法推进。

所以多引进一个指针来负责1的区域，就是下面的3指针方法。

1.三指针（题解中的双指针）

分别用三个指针标记三个颜色的区域

• zero作为标记0与1的分界线
• two标记1与2的分界线
• i负责遍历1的区间将区间内的0、2交换出去。
  

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    void sortColors(vector<int> &nums) {
        int size = nums.size();
        if (size < 2) {
            return;
        }
        // all in [0, zero) = 0
        // all in [zero, i) = 1
        // all in [two, len - 1] = 2

		 // 循环终止条件是 i == two，那么循环可以继续的条件是 i < two
        // 为了保证初始化的时候 [0, zero) 为空，设置 zero = 0，
        // 所以下面遍历到 0 的时候，先交换，再加

        int zero = 0;
        // 为了保证初始化的时候 [two, len - 1] 为空，设置 two = len
        // 所以下面遍历到 2 的时候，先减，再交换
        int two = size;
        int i = 0;
        
		// 当 i == two 上面的三个子区间正好覆盖了全部数组
        // 因此，循环可以继续的条件是 i < two
        while (i < two) {//1的区间还没有遍历完
            if (nums[i] == 0) {
                swap(nums[zero], nums[i]);
                zero++;
                i++;
            } else if (nums[i] == 1) {
                i++;
            } else {
                two--;
                swap(nums[i], nums[two]);
            }
        }
    }
};
/*
作者：liweiwei1419
链接：https://leetcode.cn/problems/sort-colors/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
*/

2.巧妙赋值

这个是题解中一个很巧妙的赋值方法！
题解链接：最简洁的双指针解法！一次遍历，无交换操作

大致思路：遍历的同时统计0和1的个数，并将数组元素首先全部赋为2。在统计过程中，每遇到1，则前n1个元素必为0或1；每当遇到0，前n0个元素必为0。于是，每当n1++，前n1个值刷成1，每当n0++，前n0个值覆盖掉之前的1赋为0。

我对代码的解读放在注释中

    public void sortColors(int[] nums) {
    	//n0记录0个个数,	n1记录0和1的个数
        int n0 = 0, n1 = 0;
        //遍历数组
        for(int i = 0; i < nums.length; i++){
            int num = nums[i];
            nums[i] = 2;	//首先将数组中所有元素赋值2
            //统计<2的个数，并将前n1个都赋为1
            if(num < 2){
                nums[n1++] = 1;
            }
            //统计=0的个数，并将前n0个都赋为0
            //由于n0<=n1，n0会将n1更新过的值覆盖掉
            if(num < 1){
                nums[n0++] = 0;
            }
        }
    }
/*
作者：颠倒的蝴蝶
链接：https://leetcode.cn/problems/sort-colors/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
*/