Leetcode75 - Sort Colors（two pointers）

题意

给定一个数组，只包含0,1,2，将0,1,2按照顺序排列。要求：
1. 只能开常数空间
2. 对数组只扫描一遍

思路

算法1

算法1是最容易想到的，我们扫描一遍数组统计一下每个数字的数目，然后再扫描一遍填进去即可。

算法2

我们可以这样想，只需要确定每个数字的开头位置即可。
首先，0的开头位置一定是0，那么，我们只需要确定1和2的开头位置即可。于是，我们设置两个指针ind1和ind2，分别代表1的开头位置，和2的开头位置的前一个数。
然后，设置一个指针i开始对数组进行扫描，并且，我们可以很容易分析出$ind1 \leq i \leq ind2$ ：

1. 若$a[i] == 0$：那么，将a[i]和a[ind1]交换位置，ind1++，i++。
2. 若$a[i] == 1$：那么，ind1不用右挪，只需要i++即可。
3. 若$a[i] == 2$：那么，我们需要将a[i]交换到后面去，因为ind2指向的是2前面的一个数，因此，我们交换a[i]和a[ind2]后，肯定ind2–，因为2的开头位置会减小，但是i不能++，因为我们指向的是2前面的一个数字，可能为1或0，因此还需要继续一轮判断。

其算法的思想类似于快排的partition。

算法3

非常巧妙的一个算法，其思想在于平移。
当我们遇到一个0的时候，所有数的起始点都要后挪一位，因此：

a[++ind2] = 2; a[++ind1] = 1; a[++ind0] = 0;

当我们遇到1的时候，只有1和2的起始点会挪，因此：

a[++ind2] = 2; a[++ind1] = 1;

当我们遇到2的时候，只有2会挪，因此：

a[++ind2] = 2;

算法4

自己的算法是设置3个指针：i, j, newi。其中，i向右扫描，j向左扫描：
当a[i] == 0时：a[newi] = 0;
当a[i] == 1时，直接i++;
当a[i] == 2时，这时候，我们要把2放后面，j指向的是2的开头位置的前一个。然后，我们再判断a[j]位置的数：
1. 若a[j] == 2：j前挪到第一个不是2的位置
2. 若a[j] == 1：直接将a[j]替换为2
3. 若a[j] == 0：将a[j]替换为2，并且a[newi]替换为0

代码

//algorithm 2, based on two index
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int i = 0, ind1 = 0, ind2 = nums.size() - 1;
        while (i <= ind2) {
            if (nums[i] == 0) swap(nums[i++], nums[ind1++]);
            else if (nums[i] == 1) i++;
            else swap(nums[i], nums[ind2--]);
        }
    }
};

//algorithm3, based on shift
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int ind1 = -1, ind2 = -1, ind3 = -1;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 0) {
                nums[++ind3] = 2;
                nums[++ind2] = 1;
                nums[++ind1] = 0;
            } 
            else if (nums[i] == 1) {
                nums[++ind3] = 2;
                nums[++ind2] = 1;
            } else {
                nums[++ind3] = 2;
            }
        }
    }
};

//algorithm3, based on index
class Solution {
public:
    void sortColors(vector<int>& a) {
        int i = 0, j = a.size() - 1, newi = 0;
        while (i <= j) {
            if (a[i] == 0) {
                a[newi++] = 0;
            } else if (a[i] == 2) {
                while (a[j] == 2 && j > i) j--;
                if (a[j] == 0) {
                    a[newi++] = 0;
                    a[j--] = 2;
                } else {
                    a[j--] = 2;
                }
            }
            i++; //i always move right
        }
        while (newi <= j) a[newi++] = 1;
    }
};