- 现将三路快排中的Partition过程代码注释如下:
//荷兰国旗问题:
// 思路:用到了partition()方法中的分三个区的思想,此时的pivot值为中间的那个元素
public class MyNetherlandsFlag {
public static int[] partition(int[] arr, int l, int r, int pivot) {
//一共分为三个区,小于区,等于区,大于区
//i指针指向小于区的右边界(i+1处为等于区),j指针指向大于区的左边界(j-1处为等于区),k指针用来遍历数组
//初始化时,默认小于区没有元素,大于区没有元素,通过不断地遍历来扩大小于区和大于区
int i = l-1;
int j = r+1;
int k = l;
while (k < j) { //循环终止条件:遍历指针遇上大于区左边界,因为k与j不可能同时都动,所以不会出现k>j的情况
if (arr[k] < pivot) {
//该元素小于pivot,将之放置在小于区域
//放置方法:将该元素与小于区域的右边界的下一个元素(i+1)交换,然后将i指针与k指针后移
//i指针后移,这便完成了小于区域的扩大
swap(arr,++i,k++);
}else if (arr[k] > pivot) {
//该元素大于pivot,将之放置在大于区域
//放置方法:将该元素与大于区域的左边界的下一个元素(j-1)交换,然后将j指针前进,m指针不变,因为此刻m处的元素还没对之判断
//j指针前进,这便完成了大于区域的扩大
swap(arr,--j,k);
}else {
//该元素等于pivot,k指针后移
k++;
}
}
//返回两个元素的数组,pivot_arr[0]是等于区左边界所在索引,pivot_arr[1]是等于区右边界所在索引
//等于区左边界索引:i+1
//等于区域右边界索引:j-1
return new int[]{i+1,j-1};
}
// for test
public static void swap(int[] arr, int i, int j) {
int tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
// for test
public static int[] generateArray() {
int[] arr = new int[10];
for (int i = 0; i < arr.length; i++) {
arr[i] = (int) (Math.random() * 3);
}
return arr;
}
// for test
public static void printArray(int[] arr) {
if (arr == null) {
return;
}
for (int i = 0; i < arr.length; i++) {
System.out.print(arr[i] + " ");
}
System.out.println();
}
public static void main(String[] args) {
int[] test = generateArray();
printArray(test);
int[] res = partition(test, 0, test.length - 1, 1);
printArray(test);
System.out.println(res[0]);
System.out.println(res[1]);
}
}