手撕代码 之 快速排序

1.实现快速排序算法

  • 问题描述

    给定一个无序数组int[ ] a,使用快速排序算法进行排序。
  • 解题思路

    对于快速排序,主要的解题思想就是分而治之。不论是从什么角度来理解,我们的方式都是在原数组中选择一个基准元素pivot,然后将小于pivot的元素全部移动到数组左侧,将大于pivot的元素全部移动到数组右侧,这样将数组分为两半,然后递归地对两侧分别继续进行排序,直到只有一个元素需要排序即可。
    为方便进行叙述,我们对数组int[ ] a进行排序,数组长度为len
    • 思考角度1: 拆东墙补西墙
      我们将基准元素从数组中提取出来,将其位置空缺。首先从右向左遍历,将比第一个pivot小的元素填入空缺 位置,然后将该位置空缺;然后从左向右遍历数组,找到第一个比pivot大的元素填入新的空缺位置,继续将该位置空缺。依次类推进行下去,直到左右指针位置发生交叉,最后将最初提取出来的元素pivot填入最终的空缺位置,并以该位置将数组分为左右两侧。代码如下。
    public void _quicksort(int[] a, int _left, int _right){
          if(_left >= _right) return;
          int left = _left, right = _right;
          int pivot = a[left]; // 最初的pivot位置空缺
          while(left < right){
              while(a[right]>=pivot && right>left)
                  right--;
              a[left] = a[right]; // 将右侧较小元素写入左侧空缺
              
              while(a[left]<=pivot && left<right)
                  left++;
              a[right] = a[left]; // 将左侧较大元素写入右侧空缺
          }
          a[right] = pivot; // 最后将pivot填入最终的空缺
          // 分割成左右两半,递归进行快速排序
          _quicksort(a, _left, right-1);
          _quicksort(a, right+1, _right);
      }
      
      public void quicksort(int[] a){
          int length = a.length;
          if(length == 0) return;
          _quicksort(a, 0, length-1);
      }
    
    • 思路2
      以上的思路已经很完整的描述了整个快速排序的思想,但是代码能不能更清晰明了一些呢?答案是肯定的。我们可以将对数组进行左右两半分割的代码提取出来写成一个函数,而这个函数的返回值正是数组左右两侧的分割位置,如下代码中的partition()
    public void _quicksort(int[] a, int left, int right){
        if(left >= right)  return;
        int x = partiton(a, left, right); // 找到左右的分割点
        _quicksort(a, left, x-1);
        _quicksort(a, x+1, right);
    }
    
    public void quicksort(int[] a){
        int length = a.length;
        if(length == 0) return;
        _quicksort(a, 0, length-1);
    }
    
    那么这个partition()函数如何实现呢?从思路1的实现中也可以得到。下边直接上代码。
    public int partition(int[] a, int _left, int _right){
        if(_left >= _right) return _left;
        int left = _left, right = _right;
        int pivot = a[left]; // 最初的pivot位置空缺
        while(left < right){
            while(a[right]>=pivot && right>left)
                right--;
            a[left] = a[right]; // 将右侧较小元素写入左侧空缺
            
            while(a[left]<=pivot && left<right)
                left++;
            a[right] = a[left]; // 将左侧较大元素写入右侧空缺
        }
        a[right] = pivot; // 最后将pivot填入最终的空缺
        reeturn right;
    }
    
  • 思路3
    对于右侧的查找分界点的过程,以上两种思路均是从左右两侧进行搜索,我们也可以采用只从一侧进行搜索的方式。

初始状态时,pivot是最左侧位置left的元素,i指向left,j指向right。

i 右侧一个元素也就是 i+1 位置开始检测,如果它的值大于pivot,那么将其与 j 位置的元素进行互换,将 j 左移一个元素 j-- ,继续对 i+1 位置进行检测;而如果它的值小于pivot,直接对 i 右移一个元素 i++

通过这种方式,使得比 pivot 大的元素全部都移到了数组的右侧,而比 pivot 小的元素全部都留在了数组左侧,实现了对数组进行分割的目的。

代码如下。

public static void quicksort_2(int[] a, int left, int right){
      if(left >= right)   return;
      int pivot = a[left];
      int i = left, j = right;
      while(i < j){
          if(a[++i]>pivot)
              swap(a, i--, j--);
      }
      swap(a, left, j);
      quicksort_2(a, left, j-1);
      quicksort_2(a, j+1, right);
}

public static void swap(int[] a, int i, int j){
      System.out.println("i: " + i + ", j: " + j);
      int temp = a[i];
      a[i] = a[j];
      a[j] = temp;
      Print(a);
}

2.快速排序拓展 之 数组中第K大(小)的元素

  • 问题描述

    给定一个无序数组int[ ] a,找到数组中第K大(小)的元素。
  • 解题思路

    不论是求第K大还是第K小,我们都可以通过 a.length-K 的方式进行转化,所以我们再次只探讨求第K小的元素。
    那么对于这个问题,最naive的想法很简单:我们只需要对数组进行从小到大进行排序,然后取出第K个元素就可以了。那么,能不能对其进行简化一下呢?
    或许我们仔细思考一下快速排序的算法过程,在每一轮的快速排序中,我们都会进行一次数组的左右分割。

    如果分割点恰好在第K个位置,那么这个位置的元素直接返回就OK了,right?
    那如果没有那么巧,分割点的位置在x(x!=K)。这种情况下,我们可以从以下两种情况来考虑:

    (1) x > K
    说明分割点左侧有超过K个数,而且这些数全部都是小于分割点位置的元素的。那么我们只需要继续从左侧继续寻找第K大的数不就可以了?

    (2) x < K
    说明分割点左侧只有x个元素不超过分割点元素,那么将这x个数刨除,我们只需要从剩下的元素中找到第(K-x)大的,就是我们要找的在整个数组中排在第K大的元素。
    代码如下,使用的分割方法是使用1.快速排序算法思路3,因为其比较简洁,是我比较喜欢的方式。

    public static void swap(int[] a, int i, int j){
          int temp = a[i];
          a[i] = a[j];
          a[j] = temp;
      }
      
      public static int getRank_K(int[] a, int left, int right, int k){
          int i = left, j = right;
          int pivot = a[left];
          while(i < j){
              if(a[++i]>pivot)
                  swap(a, i--, j--);
          }
          System.out.println("j: " + j);
          swap(a, left, j);
          if(j==k) return a[j];
          if(j < k)   return getRank_K(a, j+1, right, k); // 因为这里我们的数组下表并没有产生变化,所以这里仍然是K
          return getRank_K(a, left, j-1, k);
      }
      
      public static int rank_K(int[] a, int k){// a为非负整数数组
          int length = a.length;
          if(k >= length) return -1;
          
          return getRank_K(a, 0, length-1, k);
      }
    

3.快速排序拓展 之 数组中最大(小)的K个元素

  • 问题描述

    给定一个无序数组int[ ] a,找到数组中最大(小)的K个元素。
  • 解题思路

    不论是求第K大还是第K小,我们都可以通过 a.length-K 的方式进行转化,所以我们再次只探讨求最小的K个元素。
    对于这个问题,我们同样可以从快速排序的角度去思考:因为我们在快速排序的过程中,每次我们都会找到一个分割点,使得其左侧的元素都比它小,右侧的元素都比它大,那如果这个分割点恰好在位置(K-1),那从当前数组的起点到该位置是不是就是数组中最小的K个元素了呢?
    依照这个逻辑,再加上之前的代码,我们可以很容易地解决这个问题。
    public static void swap(int[] a, int i, int j){
      int temp = a[i];
      a[i] = a[j];
      a[j] = temp;
    }
      
    public static int partition(int[] a, int left, int right){
      int i = left, j = right;
      int pivot = a[left];
      while(i < j){
          if(a[++i]>pivot)
              swap(a, i--, j--);
      }
      swap(a, left, j);
      return j;
    }
      
    public static List<Integer> getTopK(int[] a, int k){// a为非负整数数组
      List<Integer> res = new ArrayList<Integer>();
      int length = a.length;
      if(k>length || k<0 || length==0)    return res;
      int splitIindex = partition(a, 0, length-1);
      int start = 0, end = length-1;
      while(splitIndex != k-1){
        if(splitIndex > k-1){
          end = splitIndex - 1;
          splitIndex = partition(a, start, end);
        }else{
          start = splitIndex + 1;
          splitIndex = partition(a, start, end);
        }
      }
      for(int i=0; i<k; i++){
        res.add(a[i]);
      }
      return res;
    }
    
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 201,681评论 5 474
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,710评论 2 377
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 148,623评论 0 334
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,202评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,232评论 5 363
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,368评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,795评论 3 393
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,461评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,647评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,476评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,525评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,226评论 3 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,785评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,857评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,090评论 1 258
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,647评论 2 348
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,215评论 2 341