快排的诡异之处

如果in-place做quick sort,选择放置pivot的位置会影响到代码细节。

同样的实现方式,选择第一个元素为pivot时,恢复pivot位置就得以hi为准。
public class TestQuickSort {
  public static void main(String[] args) {
    int[] arr = { 1, 4, 2, 3, 6, 5, 0 };
    quickSort(arr, 0, arr.length - 1);
    for (int i : arr) System.out.println(i);
  }
  
  public static void quickSort(int[] arr, int left, int right) {
    if (left >= right) return;
    int pivot = arr[left];
    int lo = left;
    int hi = right + 1;
    while (true) {
      while (arr[++lo] < pivot) { if (lo == right) break; }
      while (arr[--hi] > pivot) { }
      if (lo >= hi) break;
      swap(arr, lo, hi);
    }
    swap(arr, left, hi);
    quickSort(arr, left, hi - 1);
    quickSort(arr, hi + 1, right);
  }
  
  public static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
  }
}

反之,以最后一个元素为pivot时,恢复pivot就得以lo为准。
public class TestQuickSort {
  public static void main(String[] args) {
    int[] arr = { 1, 4, 2, 3, 6, 5, 0 };
    quickSort(arr, 0, arr.length - 1);
    for (int i : arr) System.out.println(i);
  }
  
  public static void quickSort(int[] arr, int left, int right) {
    if (left >= right) return;
    int pivot = arr[right];
    int lo = left - 1;
    int hi = right;
    while (true) {
      while (arr[++lo] < pivot) { }
      while (arr[--hi] > pivot) { if (hi == left) break; }
      if (lo >= hi) break;
      swap(arr, lo, hi);
    }
    swap(arr, right, lo);
    quickSort(arr, left, lo - 1);
    quickSort(arr, lo + 1, right);
  }
  
  public static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
  }
}


这是因为这种实现方式中,在选取恢复pivot的下标时,应选取“安全”的那个,也就是永远不会超过pivot所在位置的那个。
如果选取中间的元素为pivot,则无需恢复pivot,而且无论以lo还是hi为基准继续快排都行,因为在满足循环的结束条件时它们肯定相等。
public class TestQuickSort {
  public static void main(String[] args) {
    int[] arr = { 1, 4, 2, 3, 6, 5, 0 };
    quickSort(arr, 0, arr.length - 1);
    for (int i : arr) System.out.println(i);
  }
  
  public static void quickSort(int[] arr, int left, int right) {
    if (left >= right) return;
    int pivot = arr[(left + right) / 2];
    int lo = left - 1;
    int hi = right + 1;
    while (true) {
      while (arr[++lo] < pivot) { }
      while (arr[--hi] > pivot) { }
      if (lo >= hi) break;
      swap(arr, i, j);
    }
    quickSort(arr, left, lo - 1);
    quickSort(arr, lo + 1, right);
  }
  
  public static void swap(int[] arr, int i, int j) {
    System.out.println("swap " + i + " with " + j);
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
  }
}

你可能感兴趣的:(J#)