0,1,2分别代表三种颜色,要把他们放在一起
Input: [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]
void sortColors(int A[], int n) {
//三根指针,i,j,k
int j = 0, k = n-1;
//一次遍历
for (int i=0; i <= k; i++) {
//如果当前元素是0
if (A[i] == 0)
//就交换当前和从零开始的指针j所指的数,更新j
swap(A[i], A[j++]);
//如果当前是2
else if (A[i] == 2)
//就交换当前指针和指向末尾的指针k所指的数,更新k和i,这里为什么要更新i
swap(A[i--], A[k--]);
}
}