leetcode题解(三十一):75. Sort Colors

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

猜你喜欢

转载自blog.csdn.net/weixin_43869024/article/details/89448404