非递归实现快速排序

992 阅读2分钟

这是我参与11月更文挑战的第17天,活动详情查看:2021最后一次更文挑战

欢迎关注公众号OpenCoder,来和我做朋友吧~❤😘😁🐱‍🐉👀

一、栈的思路替换递归

方法的调用实际是使用了方法调用栈。递归不就是方法调用本身就是入栈和出栈的过程吗。如果是这样的话,我们就可以使用栈来替换之前的递归,在栈中存储方法每次传入的参数即可。

二、逻辑推演

例如:如下的代码我们可以使用栈表示

    public static void main(String[] args) {
        quickSort(0,10);
    }
    
    public static void quickSort(int startIndex,int endIndex){
        quickSort(0,5);
        quickSort(7,10);
    }

image-20211104192301592

三、代码

下面我们来使用代码来演示出来,假设我们需要排序的数字是:{50, 20, 30, 10, 40}

public class QuickSort {
    public static void main(String[] args) {
        int[] arr = new int[]{50, 20, 30, 10, 40};
        quickSort(arr, 0, arr.length - 1);
        System.out.println(Arrays.toString(arr));
    }
​
    public static void quickSort(int[] arr, int startIndex, int endIndex) {
        // 用一个集合栈来代替递归的函数栈
        Stack<Map<String, Integer>> sort = new Stack<Map<String, Integer>>();
        // 整个数列的起止下标,以哈希的形式入栈
        Map rootParam = new HashMap();
        rootParam.put("startIndex", startIndex);
        rootParam.put("endIndex", endIndex);
        sort.push(rootParam);
​
        // 循环结束条件:栈为空时结束
        while (!sort.isEmpty()) {
            // 栈顶元素出栈,得到起止下标
            Map<String, Integer> param = sort.pop();
            // 得到基准元素位置
            int pivotIndex = partition(arr, param.get("startIndex"), param.get("endIndex"));
            // 根据基准元素分成两部分, 把每一部分的起止下标入栈
            if (param.get("startIndex") < pivotIndex - 1) {
                Map<String, Integer> leftParam = new HashMap<String, Integer>();
                leftParam.put("startIndex", param.get("startIndex"));
                leftParam.put("endIndex", pivotIndex - 1);
                sort.push(leftParam);
            }
            if (pivotIndex + 1 < param.get("endIndex")) {
                Map<String, Integer> rightParam = new HashMap<String, Integer>();
                rightParam.put("startIndex", pivotIndex + 1);
                rightParam.put("endIndex", param.get("endIndex"));
                sort.push(rightParam);
            }
        }
    }
​
    private static int partition(int[] arr, int startIndex, int endIndex) {
        // 取第一个位置的元素作为基准元素
        int pivot = arr[startIndex];
        int left = startIndex;
        int right = endIndex;
​
        while (left != right) {
            //控制right指针比较并左移
            while (left < right && arr[right] > pivot) {
                right--;
            }
            //控制right指针比较并右移
            while (left < right && arr[left] <= pivot) {
                left++;
            }
            //交换left和right指向的元素
            if (left < right) {
                int p = arr[left];
                arr[left] = arr[right];
                arr[right] = p;
            }
        }
​
        //pivot和指针重合点交换
        int p = arr[left];
        arr[left] = arr[startIndex];
        arr[startIndex] = p;
​
        return left;
    }
}

四、总结

每循环一遍,栈顶元素都会出栈,从而排序,因为基准元素的位置所以会分成左右两部分,左右两部分再分别入栈。当栈为空时,说明排序已经完毕,退出循环,以上就是非递归方式实现快速排序。

欢迎关注公众号OpenCoder,来和我做朋友吧~❤😘😁🐱‍🐉👀