OJ - 75 对0,1,2进行排序（Sort Color）

题目：给定一个数组，数组里面的元素取值为：[0, 2]，把这个数组排序。

1.第一种做法：计数排序，因为数组里面的元素都是正数，可以直接索引到，所以我们直接可以开一个size为3的数组，用来保存0,1,2的个数。最后再直接覆写原来的数组即可。

void sortColors(vector<int>& nums) {
        vector<int> current(3, 0);
        int size = nums.size();
        for (int i = 0; i < size; ++i) {
            current[nums[i]]++;
        }
        int start = 0, end = 0;
        for (int i = 0; i < 3; ++i) {
            end += current[i];
            for (int j = start; j < end; ++j) {
                nums[j] = i;
            }
            start += current[i];
        }
    }

这个算法的时间复杂度为O(N),空间复杂度为O(max(nums))[考虑到如果输入不止为0,1,2的时候]，这个是计数排序的弊端。

我们最好弄一个只遍历一遍，并且没有额外空间的使用的算法：
我们使用三个指针来帮助我们标记当前的元素：i代表遍历数组的指针，j代表下一个0元素该出现的地方，k表示下一个2元素该出现的地方。然后我们遍历这个数组：当当前的元素为0的时候，交换i和j指针的数。并且让j自增，当当前元素为2的时候，交换i和k指针的数，并且让k自减，i自减，i自减的目的是：不知道交换过来的数到底是几，所以需要重新判断。
为什么1不用判断：因为我们处理了0和2,放在中间的1自然就有序了，不用处理。所以我们可以有下面的代码：

void sortColors(vector<int>& nums) {
        int k = nums.size() - 1;
        int j = 0;
        for (int i = 0; i <= k; i++) {
            if (nums[i] == 0) {
                swap(nums[i], nums[j++]);
            } else if (nums[i] == 2) {
                swap(nums[i--], nums[k--]);
            }
        }
    }

    void swap(int& a, int& b) {
        int c = b;
        b = a;
        a = c;
    }