三路快排中的Partition过程

• 现将三路快排中的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]);

    }
}