优化经典快排

问题1：给定一个数组arr， 和一个数num， 请把小于等于num的数放在数组的左边， 大于num的数放在数组的右边。要求额外空间复杂度O(1)， 时间复杂度O(N)。

public class a {
    public static void partition (int[] arr, int num) {
        int left = -1;
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] < num) {
                swap(arr, ++left, i);
            }
        }
    }
    public static void swap(int[] arr, int index, int position) {
        int temp = arr[index];
        arr[index] = arr[position];
        arr[position] = temp;
    }
}

问题2:（荷兰国旗问题）
给定一个数组arr， 和一个数num， 请把小于num的数放在数组的左边， 等于num的数放在数组的中间， 大于num的数放在数组的右边。要求额外空间复杂度O(1)， 时间复杂度O(N).

public class NetherlandsFlag {

	public static int[] partition(int[] arr, int l, int r, int num) {
		int less = l - 1;
		int more = r + 1;
		while (l < more) {
			if (arr[l] < num) {
				swap(arr, ++less, l++); // 将当前与小于区域的下个位置交换
			} else if (arr[l] > num) {
				swap(arr, --more, l);  // 大于区域的前一个数与当前交换
			} else {  // == num
				l++;
			}
		}
		return new int[] { less + 1, more - 1 };  // less+1是等于区域的第一个位置，more-1是等于区域的最后一个位置。
	}
}

问题3：改进后快排

• 改进1：使用随机选取划分点，避免极端情况（已经有序）。
• 改进2：使用荷兰国旗问题优化快排，使等于划分数不再继续参与划分，避免无效排序。

public class QuickSort {

	public static void quickSort(int[] arr) {
		if (arr == null || arr.length < 2) {
			return;
		}
		quickSort(arr, 0, arr.length - 1);
	}

	public static void quickSort(int[] arr, int l, int r) {
		if (l < r) {
			swap(arr, l + (int) (Math.random() * (r - l + 1)), r);  //随机快排，避免最坏情况
			int[] p = partition(arr, l, r);  // 划分过程，得到等于划分数的左右边界
			quickSort(arr, l, p[0] - 1);  // 继续左边
			quickSort(arr, p[1] + 1, r);  // 继续右边
		}
	}

	public static int[] partition(int[] arr, int l, int r) {  // 以arr[r]数字划分
		int less = l - 1;
		int more = r;
		while (l < more) {
			if (arr[l] < arr[r]) {
				swap(arr, ++less, l++);
			} else if (arr[l] > arr[r]) {
				swap(arr, --more, l);
			} else {
				l++;
			}
		}
		swap(arr, more, r); // more是指向大于区域的第一个大于arr[r]的数，最后需要将划分数调整到划分位置。
		return new int[] { less + 1, more };  // 返回中间等于区域的首尾边界
	}

	public static void swap(int[] arr, int i, int j) {
		int tmp = arr[i];
		arr[i] = arr[j];
		arr[j] = tmp;
	}
}