三色排序(荷兰国旗问题)

class ThreeColor {
public:
    vector<int> sortThreeColor(vector<int> A, int n) {
        // write code here
        int left=0;
        int right=n-1;
        for(int i=0;i<n&&i<=right;i++)
        {
            if(A[i]==1)
            {
                continue;
            }
            else if(A[i]==0)
            {
                swap(A[i],A[left]);
                left++;
            }
            else
            {
                swap(A[i],A[right]);
                i--;
                right--;
            }
        }
        return A;
    }
};