C++三色旗问题

class Solution {
public:
	void holandFlag(int a[], int n) {
		int i = -1;//令i为数组中第一个元素之前的一个元素
		int j = n;//令j为数组中最后一个元素之后的一个元素
		int index = 0;//index为数组中第一个元素的下标，用它来遍历数组
		while (index<j)//循环结束条件，此方法不用遍历全部的数组元素，当index和j相遇说明整个数组已经排列好
		{
//如果index下标处的值为0，则先将++i，然后交换，交换完成之后将index++
			if (a[index] == 0) {
				swap(a[++i], a[index]);
				index++;
			}
//如果index下标处的值为2，则先将--j，然后交换，交换完成之后index不++，因为无法保证后边的数，是否符合我们的排序要求
			else if (a[index] == 2)
			{
				swap(a[--j], a[index]);
			}
//如果index下标处的值为1，将index++
			else
			{
				index++;
			}
		}
	}
//打印数组
	void print(int a[],int n) {
		for (int i = 0; i < n; i++) {
			cout << a[i] << "  ";
		}
		cout << endl;
	}
};
int main() {
	Solution s;
	int a[] = { 1,2,0,2,1,1,0,0,2,1 };
	s.holandFlag(a, 10);
	s.print(a, 10);
	return 0;
}