LeetCode刷题---双指针---75

75. 颜色分类 - 力扣（LeetCode）

题目

给定一个包含红色、白色和蓝色、共 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]

提示：

• n == nums.length
• 1 <= n <= 300
• nums[i] 为 0、1 或 2

自己的思路和代码

思路：

        原地排序就是不开辟新的空间，在给出的数据空间进行排序。很简单的一道题。

        扫描两次即可：

        第一次遇到12就往后找0的位置交换
        第二次遇到2就往后找1的位置交换

代码：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        for(int i=0; i<nums.size() - 1; i++) { 
            if(nums[i] == 1 || nums[i] == 2) {
                bool flag = false;
                for(int j=i + 1; j<nums.size(); j++) {
                    if(nums[j] == 0) {
                        nums[j] = nums[i];
                        nums[i] = 0;
                        break;
                    } else if (j == nums.size() - 1) {
                        flag = true;
                    }
                }
                if(flag) break;
            }
        }
        for(int i=0; i<nums.size() - 1; i++) { 
            if(nums[i] == 2) {
                bool flag = false;
                //printf("%d\n", i);
                for(int j=i + 1; j<nums.size(); j++) {
                    if(nums[j] == 1) {
                        nums[j] = nums[i];
                        nums[i] = 1;
                        break;
                    } else if (j == nums.size() - 1) {
                        flag = true;
                    }
                }
                if(flag) break;
            }
        }
    }
};