Leetcode 75. 颜色分类

题目

Leetcode 75. 颜色分类

代码（首刷看解析）

双指针

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0 = 0, p1 = 0;
        int n = nums.size();
        for(int i = 0; i < n; i++) {
            if(nums[i] == 0) {
                swap(nums[i], nums[p0]);
                if(p0 < p1) {   // 如果p0<p1，会把1交换出去，所以需要将i换到p1的位置
                    swap(nums[i], nums[p1]);    
                }
                p1++;
                p0++;
            } else if(nums[i] == 1) {
                swap(nums[i], nums[p1]);
                p1++;
            }
        }
    }
};

代码（9.14 二刷自解）

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