[Leetcode][Sort][Sort Colors]

很重要的一道题，考察的是快排的partition思路。

1、先写了个快排练手，果然很久没写了。。有好几个BUG。。。

class Solution {
public:
    int partition(int objects[], int start, int end) {
        int pivot = objects[start];
        while (start < end) {
            while (start < end && objects[end] >= pivot) {end--;}
            if (start < end) {
                objects[start] = objects[end];
            }
            while (start < end && objects[start] <= pivot) {start++;}
            if (start < end) {
                objects[end] = objects[start];
            }
        }
        objects[start] = pivot;
        return start;
    }
    void quickSort(int objects[], int start, int end) {
        if (start >= end) return;
        int pivot = partition(objects, start, end);
        quickSort(objects, start, pivot - 1);
        quickSort(objects, pivot + 1, end);
    }
    void sortColors(int A[], int n) {
        quickSort(A, 0, n - 1);
    }
};

快排的时间复杂度是O(n*lgn)，空间复杂度是O(1)

 

2、更简单的是记录一下0,1,2的个数。。。就像题目里所说的一样。。

关键是如何想出来一种one-pass的方法，只需要遍历一次即可！其实每个元素还是访问了好多次。

class Solution {
public:
    void sortColors(int A[], int n) {
        int left = 0;
        int right = n - 1;
        int i = 0;
        while(i <= right) {
            if (A[i] == 0 && i != left) {
                swap(A[left++], A[i]);
            } else if (A[i] == 2 && i != right) {
                swap(A[right--], A[i]);
            } else {
                i++;
            }
        }
    }
};

代码很简单，边界想了很多次，忧桑~

 

3、用Paritition做，可以扩展到n种颜色。

看到了一篇文章http://www.csse.monash.edu.au/~lloyd/tildeAlgDS/Sort/Flag/ ，说的挺清楚的。于是模仿着又写了一遍。

其实本质是一样的，用2个指针来标记范围，一个指针不停的往后移动。

其实不太确定怎么用。需要进一步深入研究！！