Leetcode#75 Sort Colors

原题地址

w指针遍历数组，如果w跨过b，则退出

若w指向的元素是白色，指针右移

若w指向的元素是红色，交换w和r指向的元素

若w指向的元素是蓝色，交换w和b指向的元素

代码：

void sortColors(int A[], int n) {
  int r = 0;
  int w = 0;
  int b = n - 1;

  while (w < n) {
    switch (A[w]) {
    case 0:
      if (w > r)
        swap(A[r], A[w]);
      else
        w++;
      r++;
      break;
    case 1:
      w++;
      break;
    case 2:
      if (w < b)
        swap(A[b], A[w]);
      else
        w++;
      b--;
      break;
    default:
      break;
    }
  }
}