[M双指针] lc75. 颜色分类(双指针+思维+脑筋急转弯)

1. 题目来源

链接：75. 颜色分类

2. 题目解析

前导题：

• [M双指针] lc1712. 将数组分成三个子数组的方案数(双指针+前缀和+周赛222_3)

采用三指针的思想，将区间分为三段。且考虑边界情况及枚举的选择，本题和前导题都是非常经典的三指针问题！

现在一般三指针来做都是需要划分区间，枚举 i 再在 0~i-1 之间枚举 j，再在后面枚举 k，这样的顺序来枚举三个指针。且区间划分是 0~j-1 为第一段区间，j~k-1 为第二段区间，k~n-1 为第三段区间。这个思想及边界情况的考虑和设计都是很巧妙的！！！

体悟算法之美！

• 时间复杂度：$O(n)$。
• 空间复杂度：$O(1)$

代码：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        for (int i = 0, j = 0, k = n - 1; i <= k; ) {
            if (nums[i] == 0) swap(nums[i], nums[j]), i ++, j ++ ;
            else if (nums[i] == 1) i ++ ;
            else if (nums[i] == 2) swap(nums[i], nums[k]), k -- ;
        }   
    }
};

---

20210718 近期偶然看到面经，是百度二面的笔试题。

算法很不错，思维题。题解区给出了很不错的算法，可以看看官方题解，力荐！

两次遍历：第一次遍历将 0 直接放到首位，并记录放置 0 的个数。第二次遍历，将 1 开始顺序交换。最后剩下的就是 2 全在最后了。

一次遍历：除过上式的方法，官方还给了一种方法，写的有点多，懒得看了。