V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
mightofcode
V2EX  ›  Java

简明快速排序写法

  •  
  •   mightofcode · 2020-01-04 20:52:48 +08:00 · 3142 次点击
    这是一个创建于 1784 天前的主题,其中的信息可能已经有所发展或是发生改变。

    快速排序代码有几个坑需要注意:
    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());
        }
    }
    
    
    7 条回复    2020-01-06 09:43:18 +08:00
    justyy
        1
    justyy  
       2020-01-05 07:42:34 +08:00
    mightofcode
        2
    mightofcode  
    OP
       2020-01-05 09:30:44 +08:00
    @justyy 学习了,你贴的代码比较简洁
    不过我更喜欢可读性更好(但是更复杂)的版本
    hobochen
        3
    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]
    ```
    我咋没觉得短会影响可读性呢。。
    jorneyr
        5
    jorneyr  
       2020-01-06 09:20:36 +08:00
    大多数的快排用的是双边循环法,单边循环法的快速排序最简洁易懂
    mightofcode
        6
    mightofcode  
    OP
       2020-01-06 09:41:49 +08:00
    @jorneyr 学到了,单边循环法太巧妙了
    mightofcode
        7
    mightofcode  
    OP
       2020-01-06 09:43:18 +08:00
    @hobochen haskell 很优美
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   2355 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 25ms · UTC 15:52 · PVG 23:52 · LAX 07:52 · JFK 10:52
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.