Java与算法之(2) - 快速排序

Z时代
2024-01-10
分类：综合

java

快速排序的基本思路是，每次选定数列中的一个基准数，将小于基准数的数字都放到基准数左边，大于基准数的数字都放到基准数右边。然后再分别对基准数左右的两个数列分别重复以上过程。仍以4 3 6 2 7 1 5为例。
选定最左侧数字4为基准数，首先从右开始向左找小于4的数，找到第一个数1后停止。然后从左开始向右找到第一个大于4的数，即6。

交换这两个数的位置，得到

继续寻找，仍然从右边开始，从上一步找到1的位置向左寻找小于4的数，找到2停止。然后从左边找到6的位置向右找大于4的数。右移一格后，和右侧来的“探路者”相遇了，这意味着这一轮排序结束。

最后把结束位置的数和基准数交换

观察完成后的数列，可以看到以基准数4为分界线，左边的数全都比4小，右边的数全都比4大。接下来分别对左边的2 3 1和右边的7 6 5重复上面的排序步骤。

2 3 1
以2为基准数 -> 2 1 3 -> 1 2 3

7 6 5
以7为基准数 -> 5 6 7
我们例子中的数字较少，如果数列足够长，对第一次排序后得到的子数列排序，将再得到两个子数列，然后再一分为二、二分为四。。。直到以基准数拆分后两边都只剩下一个数字。首先来看递归形式的实现代码

public class QuickSort {
    public void sort(int left, int right, int... numbers) {
        if (left >= right) {
            return;
        }
        int temp = numbers[left];
        int t = 0;
        int i = left;
        int j = right;
        while (i != j) {
            // 先从右往左找
            while (numbers[j] >= temp && i < j)
                j--;
            // 再从左往右找
            while (numbers[i] <= temp && i < j)
                i++;
            // 交换两个数在数组中的位置
            if (i < j) {
                t = numbers[i];
                numbers[i] = numbers[j];
                numbers[j] = t;
            }
        }
        // 将基准数归位
        numbers[left] = numbers[i];
        numbers[i] = temp;
        sort(left, i - 1, numbers);
        sort(i + 1, right, numbers);
    }
}

测试代码

    public static void main(String[] args) {
        int[] numbers = new int[] { 4, 3, 6, 2, 7, 1, 5 };
        new QuickSort().sort(0, numbers.length - 1, numbers);
        System.out.print("after: ");
        for (int i = 0; i < numbers.length; i++) {
            System.out.print(numbers[i] + "  ");
        }
        System.out.println();
    }

输出

after: 1  2  3  4  5  6  7

另一种实现方式是使用栈代替递归

    public void sortWithoutRecursion(int left, int right, int... numbers) {
        LinkedList<Integer> stack = new LinkedList<>();
        int index;
        stack.push(left);
        stack.push(right);
        while (!stack.isEmpty()) {
            right = stack.pop();
            left = stack.pop();
            index = partition(left, right, numbers);
            if (left < index - 1) {
                stack.push(left);
                stack.push(index - 1);
            }
            if (right > index + 1) {
                stack.push(index + 1);
                stack.push(right);
            }
        }
    }
    public int partition(int left, int right, int... numbers) {
        int temp = numbers[left];
        while (left < right) {
            while (numbers[right] >= temp && left < right)
                right--;
            numbers[left] = numbers[right];
            while (numbers[left] <= temp && left < right)
                left++;
            numbers[right] = numbers[left];
        }
        numbers[left] = temp;
        return left;
    }