颜色分类(数组专题)

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

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

进阶:

你可以不使用代码库中的排序函数来解决这道题吗?
你能想出一个仅使用常数空间的一趟扫描算法吗?
 

示例 1:

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

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

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

思路:单指针方法,我们可以设置一个ptr变量来指向数组的首位置,再用一个指针去遍历数组,如果找到0,就把它和ptr的值交换,然后ptr+1.再遍历一次,找1,如果找到1,就和ptr的交换,这样两次遍历后,就能把0和1都放到数组的前面了。

代码:

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int ptr = 0;
        int n = nums.size();
        for(int j = 0; j < n; j++){
            if(nums[j] == 0){
                swap(nums[ptr],nums[j]);
                ptr++;
            }
        }
        for(int j = 0; j < n; j++){
            if(nums[j] == 1){
                swap(nums[ptr],nums[j]);
                ptr++;
            }
        }
    }
};
原文地址:https://www.cnblogs.com/YXBLOGXYY/p/14255698.html