快速排序--荷兰国旗三色优化

快速排序作为十大经典排序之一，大家应该有所耳闻，对它的优化网上也有不少，比如随机数快排，中位数快排，甚至有的走火入魔去融合插入排序或多线程……今天我们不成魔，做个优化针对于重复的数字，普通快排一次partition归位一个数，经过优化后直接归位所有和它相同的数，即整个区间划分为<num，==num，>num的三区间，就像荷兰国旗那样所以就叫这个了。

---

具体思路：划分为3个区，小于区，等于区，大于区。i指针从左往右循环，如果i位置数比候选num小，就把小于区右边一位与它交换，小于区右扩，i++；反之比它大，就把大于区左边一个与它交换，但i停在原地（因为交换后的数还不知道大小，需要再次比较）；如果相等，则直接i++；当指针i撞上了大于区第一个数，就终止了。看下图应该好理解：

 

 

---

不管是大于num还是小于num，都会让小于区往右扩，或大于区往左扩，指针i与小于区之间则是等于区，当i遇到从右往左移动的大于区的时候，就完成了一次partition过程。一次partition把所有num相等的全排完序，并返回它们的下标。java版的代码如下：

---

package class03;

import java.util.Arrays;

/**
 * @Description
 * @Package: class03
 * @Author: Ray
 * @CreateTime: 2020/7/5 15:07
 * @E-mail: 634302021@qq.com
 */
public class Test11 {
	
	/**
	 * 荷兰国旗partition
	 *
	 * @param arr
	 * @param left
	 * @param right
	 * @return 返回数组里有2个值：[等于区域的左边界下标，等于区域右边界下标]
	 */
	public static int[] netherlandsFlag(int[] arr, int left, int right) {
		if (left == right) {    //只有一个数的情况
			return new int[]{left, right};
		}
		int i = left;   //指针i
		int small = left - 1;   //小于区初始位置为left左边一位
		int big = right + 1;    //大于区初始位置为right右边一位
		int num = arr[right];   //取最右的数作为候选数
		while (i < big) {   //i遇到大于区才终止
			if (num == arr[i]) {    //等于的逻辑
				i += 1;
			} else if (arr[i] > num) {  //大于的逻辑，注意交换后，i停在原地等候下一循环比较
				swap(arr, big - 1, i);
				big -= 1;
			} else {    //小于的逻辑
				swap(arr, small + 1, i);
				small += 1;
				i += 1;
			}
		}
		return new int[]{small + 1, big - 1};   //返回等于区的范围下标
	}
	
	/**
	 * 递归过程
	 * @param arr
	 * @param l
	 * @param r
	 */
	public static void process(int[] arr, int l, int r) {
		if (l < r) {
			swap(arr, l + (int) (Math.random() * (r - l + 1)), r); //生成l~r范围的随机数，并与r位置做交换，避免最坏时间复杂度O(n^2)出现
			int[] equalSpan = netherlandsFlag(arr, l, r);   //等于区域
			process(arr, l, equalSpan[0] - 1);    //从l到等于区左一位递归
			process(arr, equalSpan[1] + 1, r);    //从等于区右一位到r递归
		}
	}
	
	/**
	 * 快速排序
	 * @param arr
	 */
	public static void quickSort(int[] arr) {
		if (arr == null || arr.length < 2) {
			return;
		}
		process(arr, 0, arr.length - 1);
	}
	
	/**
	 * 交换
	 * @param arr
	 * @param left
	 * @param right
	 */
	public static void swap(int[] arr, int left, int right) {
		int tmp = arr[left];
		arr[left] = arr[right];
		arr[right] = tmp;
	}
	
	public static void main(String[] args) {
		int[] arr = {1, 5, 4, 3, 2, 7, 8, 9, 4, 7, 7, 8, 7};
		quickSort(arr);
		System.out.println(Arrays.toString(arr));
		
	}
}