Sort Colors

void sortColors(int* nums, int numsSize) {
　　if(numsSize<2||nums==NULL)
　　　　return;
　　int *ptr=malloc(3*sizeof(int));
　　int i,j;
　　for(i=0;i<3;i++)//初始化
　　{
　　　　ptr[i]=0;
　　}
　　for(i=0;i<numsSize;i++)//对三种颜色计数
　　{
　　　　ptr[nums[i]]++; 
　　}

　　for(i=0;i<ptr[0];i++)　　{
　　　　nums[i]=0;
　　}
　　j=ptr[0]+ptr[1];
　　for(;i<j;i++)
　　{
　　　　nums[i]=1;
　　}
　　　j+=ptr[2];
　　for(;i<j;i++)
　　{
　　　　nums[i]=2;
　　}
}

 1 // LeetCode, Sort Colors
 2 // Counting Sort
 3 // 时间复杂度O(n)，空间复杂度O(1)
 4 class Solution {
 5 public:
 6 　　void sortColors(int A[], int n) {
 7 　　int counts[3] = { 0 }; // 记录每个颜色出现的次数
 8　　 for (int i = 0; i < n; i++)
 9 　　　　counts[A[i]]++;
10 　　for (int i = 0, index = 0; i < 3; i++)
11 　　　　for (int j = 0; j < counts[i]; j++)
12 　　　　　　A[index++] = i;
13 　　}
14 };

 1 // LeetCode, Sort Colors
 2 // 双指针，时间复杂度O(n)，空间复杂度O(1)
 3 class Solution {
 4 public:
 5     void sortColors(int A[], int n) {
 6     // 一个是red 的index，一个是blue 的index，两边往中间走
 7         int red = 0, blue = n - 1;
 8         for (int i = 0; i < blue + 1;) {
 9 　　　　　　if (A[i] == 0)
10 　　　　　　　　swap(A[i++], A[red++]);
11　　　　　　 else if (A[i] == 2)
12 　　　　　　　　swap(A[i], A[blue--]);
13 　　　　　　else
14 　　　　　　　　i++;
15 　　　　}
16 　　}
17 };            

// LeetCode, Sort Colors
// 重新实现partition()
// 时间复杂度O(n)，空间复杂度O(1)
class Solution {
　　public:
　　　　void sortColors(int A[], int n) {
　　　　　　partition(partition(A, A + n, bind1st(equal_to<int>(), 0)), A + n,bind1st(equal_to<int>(), 1));
　　　　}
　　private:
　　　　template<typename ForwardIterator, typename UnaryPredicate>
　　　　ForwardIterator partition(ForwardIterator first, ForwardIterator last,UnaryPredicate pred) {
　　　　　　auto pos = first;
　　　　　　for (; first != last; ++first)
　　　　　　　　if (pred(*first))
　　　　　　　　　　swap(*first, *pos++);
　　　　　　return pos;
　　　　}
};