【Leetcode 75】颜色分类 ——双指针-CSDN博客

觉得此文有帮助的，就点个赞或留个言呐~ 若要转载的话，请注明文章出处哦

本文链接：https://blog.csdn.net/why_still_confused/article/details/136160137

文章介绍了如何使用双指针技巧对一个包含红、白、蓝三种颜色元素的整数数组进行排序，要求相同颜色的元素相邻且遵循特定顺序。通过两个指针p0和p2，一次遍历即可完成排序过程。

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

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至数组前端，第二次遍历替换所有1至所有0的末尾。

如果要在一次遍历中完成排序，我们可以使用双指针解决本题，指针p0从数组首开始，指针p2从数组末尾开始，遍历数组的过程中将2替换至末尾，然后将0替换至数组头，即可完成排序

双指针，顾名思义就是同时使用两个指针，在序列、链表结构上指向的是位置，在树、图结构中指向的是节点，通过或同向移动，或相向移动来维护、统计信息

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length, p0 = 0, p2 = n - 1;
        for(int i = 0; i <= p2; i++){
            while(i <= p2 && nums[i] == 2){
                swap(nums, i, p2);
                p2--;
            }
            if(nums[i] == 0){
                swap(nums, i, p0);
                p0++;
            }
        }

    }

    private void swap(int[] nums, int i, int j){
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}