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;
}
};
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;
}
};