LeetCode 75. 颜色分类（三指针，双指针）-优快云博客

本文链接：https://blog.youkuaiyun.com/virtual_YL/article/details/129629896

文章介绍了如何解决LeetCode75题——颜色分类，主要使用了三指针策略。通过i、j、t三个指针分别处理0、1、2三种颜色的边界，确保数组中0在前，1在中，2在后。文中给出了两种不同的实现方式，第一种方式详细解释了指针移动的逻辑，第二种方式则更简洁。通过代码示例和分析，帮助读者理解排序过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LeetCode 75. 颜色分类

原题
思路
代码
运行截图
收获

原题

LeetCode 75. 颜色分类

思路

用三指针排序
指针i指向0的末尾，j指向2的前端，t指向1的末尾，k是当前遍历的数。[k,j]是还未处理的区间，[0, k）是已经处理好的0， 1排序好的区间，(j, n-1]是只有2的区间。
如果当前遍历的数等于0，把第k个数和第i个数交换，为什么此时i和t都能往后移呢，如果k前都是0，那么i和t肯定都往后移，但如果k前是0，0，1，1这种配置，那么肯定是1被转移到了k目前的位置，所以1的末尾位置t也要往后移一个。
如果当前遍历的数等于2，那就将第k个数和第j个数交换，此时k不往后移，因为还得判断换过来的k是0还是1
如果当前遍历的数是1，那么t++，k++就好

代码

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

官方题解方法：其实感觉大致思想是一致的，不过确实写得简单明了多了

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