简明快速排序写法

2020-01-04 20:52:48 +08:00
 mightofcode

快速排序代码有几个坑需要注意:
1,partition 的过程比较复杂,很容易出错,某些行数很短的代码很简洁但是不容易看懂,不如采用简单直观的方法:用第一个元素作为轴、partition 结束之后再把轴 swap 到中间
2,选轴如果固定选第一个,可能进入 worst case,复杂度退化为 N2,使用随机选轴来解决这个问题
3,如果有大量重复元素,快速排序同样会陷入 N2,此时需要使用三向快速排序:将数组分为小于、等于、大于轴的三部分,我们可以通过两步 partition 来完成:首先分为小于等于、大于的两部分,然后再进行一次 partition,拆分成三部分。

package com.mocyx.algs.sort;

import java.util.Random;
import java.util.Scanner;

/**
 * @author Administrator
 */
public class FastSort {
    static int[] arrs;
    static void swap(int a, int b) {
        int v = arrs[a];
        arrs[a] = arrs[b];
        arrs[b] = v;
    }

    static Random random = new Random(System.currentTimeMillis());

    static void choosePovit(int s, int e) {
        int m = s + random.nextInt(e - s);
        swap(s, m);
    }

    /**
     * @param s
     * @param e
     * @param leftEqual true 表示把相等的放在左边,false 表示把相等的放在右边
     * @return
     */
    static int partition(int s, int e, boolean leftEqual) {

        int i = s + 1;
        int j = e;
        int pivotValue = arrs[s];
        while (true) {
            if (leftEqual) {
                while (i <= e && arrs[i] <= pivotValue) {
                    i += 1;
                }
                while (j >= s + 1 && arrs[j] > pivotValue) {
                    j -= 1;
                }
            } else {
                while (i <= e && arrs[i] < pivotValue) {
                    i += 1;
                }
                while (j >= s + 1 && arrs[j] >= pivotValue) {
                    j -= 1;
                }
            }
            if (i <= e && j >= s + 1 && i <= j) {
                swap(i, j);
                i += 1;
                j -= 1;
            }else {
                break;
            }
        }
        swap(s, i - 1);
        return i - 1;
    }

    static void sort(int s, int e, int depth) {
        if (s >= e) {
            return;
        }
        //随机选择轴,避免进入 worst case
        choosePovit(s, e);
        //第一次 partition,小于等于轴的置换到轴右边
        int mr = partition(s, e, true);
        //第二次 partition,把等于轴的置换到右侧
        swap(s, mr);
        int ml = partition(s, mr, false);
        //现在数组分为三块:小于轴 等于轴 大于轴的
        sort(s, ml - 1, depth + 1);
        sort(mr + 1, e, depth + 1);
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int c = scanner.nextInt();
        arrs = new int[c];
        for (int i = 0; i < c; i++) {
            arrs[i] = scanner.nextInt();
        }
        sort(0, arrs.length - 1, 1);
        StringBuilder stringBuilder = new StringBuilder();
        for (int i = 0; i < arrs.length; i++) {
            stringBuilder.append(String.format("%d ", arrs[i]));
        }
        System.out.print(stringBuilder.toString());
    }
}

3031 次点击
所在节点    Java
7 条回复
justyy
2020-01-05 07:42:34 +08:00
mightofcode
2020-01-05 09:30:44 +08:00
@justyy 学习了,你贴的代码比较简洁
不过我更喜欢可读性更好(但是更复杂)的版本
hobochen
2020-01-05 20:31:39 +08:00
```haskell
qsort [] = []
qsort (x:xs) = qsort [y | y <- xs, y < x] ++ [x] ++ qsort [y | y <- xs, y >= x]
```
我咋没觉得短会影响可读性呢。。
KentY
2020-01-06 05:23:00 +08:00
jorneyr
2020-01-06 09:20:36 +08:00
大多数的快排用的是双边循环法,单边循环法的快速排序最简洁易懂
mightofcode
2020-01-06 09:41:49 +08:00
@jorneyr 学到了,单边循环法太巧妙了
mightofcode
2020-01-06 09:43:18 +08:00
@hobochen haskell 很优美

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/635105

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX