leetcode 日经贴，Cpp code -Sort Colors

Sort Colors

class Solution {
public:
    void swap(int &a, int &b) {
        int c = a;
        a = b;
        b = c;
    }
    void sortColors(int A[], int n) {
        const int red = 0, white = 1, blue = 2;
        int st = 0, ed = n;
        for (int i = 0; i < ed; ++i) {
            if (A[i] == red) {
                swap(A[st++], A[i]);
            } else if (A[i] == blue) {
                swap(A[i--], A[--ed]);
            }
        }
    }
};