荷兰国旗问题与快速排序算法
作者:Grey
原文地址:
荷兰国旗问题
问题描述
给定一个整数数组,给定一个值 K,这个值在原数组中一定存在,要求把数组中小于 K 的元素放到数组的左边,大于K的元素放到数组的右边,等于 K 的元素放到数组的中间。
时间复杂度要求 O ( N ) O(N) O(N),空间复杂度要求 O ( 1 ) O(1) O(1)。
主要思路
设置两个变量l
和r
,其中小于i
位置的值都是比 K 小的数,
i
…
r
i…r
i…r都是等于 K 的数,大于r
位置的值都是大于 K 的数。
初始值
l
=
−
1
,
r
=
a
r
r
.
l
e
n
g
t
h
;
l = - 1, r = arr.length;
l=−1,r=arr.length; 表示都还没考察过数组的任何一个元素,然后开始遍历数组,遍历到的位置为i
,arr[i]
有三种情况
情况 1
a r r [ i ] > K arr[i] > K arr[i]>K
情况 2
a r r [ i ] = = K arr[i] == K arr[i]==K
情况 3
a r r [ i ] < K arr[i] < K arr[i]<K
对于情况 1, 只需要将i
位置的值和
r
−
1
r - 1
r−1位置的值交换,然后r--,i++
;
对于情况 3, 只需要将i
位置的值和
l
+
1
l + 1
l+1位置的值交换,然后l++,i++
;
对于情况 2, 只需要将i
位置的值和
r
−
1
r-1
r−1位置的值交换,然后i++
。
数组遍历完毕后,原始数组就形成了:小于 K 的元素放到数组的左边,大于 K 的元素放到数组的右边,等于 K 的元素放到数组的中间。
完整代码见
class Solution {
public static void sortColors(int[] arr) {
int l = -1;
int r = arr.length;
int i = 0;
while (i < r) {
if (arr[i] > 1) {
swap(arr, i, --r);
} else if (arr[i] < 1) {
swap(arr, i++, ++l);
} else {
// arr[i] == 1
i++;
}
}
}
private static void swap(int[] arr, int l, int r) {
if (l == r) {
return;
}
arr[l] = arr[l] ^ arr[r];
arr[r] = arr[l] ^ arr[r];
arr[l] = arr[l] ^ arr[r];
}
}
快速排序
基于上述荷兰国旗算法的原型,我们可以实现快速排序算法,流程是
在 a r r [ L … R ] arr[L…R] arr[L…R]范围上,进行快速排序的过程如下
-
在这个范围上,随机选一个数记为
num
, -
用
num
对该范围使用荷兰国旗算法, < n u m < num <num 的数在左部分, = = n u m == num ==num 的数在中间, > n u m > num >num 的数在右部分。假设 = = n u m == num ==num的数所在范围是 [ a , b ] [a,b] [a,b] -
对 a r r [ L . . . a − 1 ] arr[L...a-1] arr[L...a−1]进行快速排序(递归)
-
对 a r r [ b + 1... R ] arr[b+1...R] arr[b+1...R]进行快速排序(递归)
因为每一次荷兰国旗算法都会搞定一批数的位置且不会再变动,所以排序能完成
-
通过分析知道,划分值越靠近中间,性能越好;越靠近两边,性能越差
-
随机选一个数进行划分的目的就是让好情况和差情况都变成概率事件
-
把每一种情况都列出来,会有每种情况下的时间复杂度,但概率都是 1 / N 1/N 1/N
-
那么所有情况都考虑,时间复杂度就是这种概率模型下的长期期望。
时间复杂度 O ( N ∗ l o g N ) O(N*logN) O(N∗logN),额外空间复杂度 O ( l o g N ) O(logN) O(logN)都是这么来的。
题目链接:LintCode 464 · Sort Integers II
完整代码见
public class Solution {
public static void sortIntegers2(int[] arr) {
if (null == arr || arr.length < 2) {
return;
}
process(arr, 0, arr.length - 1);
}
public static void process(int[] arr, int s, int e) {
if (s >= e) {
return;
}
swap(arr, e, s + (int) (Math.random() * (e - s)));
int[] range = sortColors(arr, s, e);
process(arr, s, range[0] - 1);
process(arr, range[1] + 1, e);
}
public static void swap(int[] arr, int i, int j) {
if (i == j) {
return;
}
arr[i] = arr[i] ^ arr[j];
arr[j] = arr[i] ^ arr[j];
arr[i] = arr[i] ^ arr[j];
}
public static int[] sortColors(int[] arr, int s, int e) {
int l = s - 1;
int r = e + 1;
int p = arr[e];
int i = s;
while (i < r) {
if (arr[i] > p) {
swap(arr, i, --r);
} else if (arr[i] < p) {
swap(arr, i++, ++l);
} else {
i++;
}
}
return new int[]{l + 1, r - 1};
}
}