嘿,還記得那些年遺忘的排序演算法嗎

上天眷顧我發表於2016-10-24

用Objective-C實現幾種基本的排序演算法,並把排序的過程圖形化顯示。其實演算法還是挺有趣的 ^ ^.

  • 選擇排序
  • 氣泡排序
  • 插入排序
  • 快速排序

選擇排序

以升序為例。
選擇排序比較好理解,一句話概括就是依次按位置挑選出適合此位置的元素來填充。

  1. 暫定第一個元素為最小元素,往後遍歷,逐個與最小元素比較,若發現更小者,與先前的"最小元素"交換位置。達到更新最小元素的目的。
  2. 一趟遍歷完成後,能確保剛剛完成的這一趟遍歷中,最的小元素已經放置在前方了。然後縮小排序範圍,新一趟排序從陣列的第二個元素開始。
  3. 在新一輪排序中重複第1、2步驟,直到範圍不能縮小為止,排序完成。

選擇排序.gif

以下方法在NSMutableArray+JXSort.m中實現

- (void)jx_selectionSortUsingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    if (self.count == 0) {
        return;
    }
    for (NSInteger i = 0; i < self.count - 1; i ++) {
        for (NSInteger j = i + 1; j < self.count; j ++) {
            if (comparator(self[i], self[j]) == NSOrderedDescending) {
                [self jx_exchangeWithIndexA:i indexB:j didExchange:exchangeCallback];
            }
        }
    }
}

氣泡排序

  1. 在一趟遍歷中,不斷地對相鄰的兩個元素進行排序,小的在前大的在後,這樣會造成大值不斷沉底的效果,當一趟遍歷完成時,最大的元素會被排在後方正確的位置上。
  2. 然後縮小排序範圍,即去掉最後方位置正確的元素,對前方陣列進行新一輪遍歷,重複第1步驟。直到範圍不能縮小為止,排序完成。

氣泡排序.gif
- (void)jx_bubbleSortUsingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    if (self.count == 0) {
        return;
    }
    for (NSInteger i = self.count - 1; i > 0; i --) {
        for (NSInteger j = 0; j < i; j ++) {
            if (comparator(self[j], self[j + 1]) == NSOrderedDescending) {
                [self jx_exchangeWithIndexA:j indexB:j + 1 didExchange:exchangeCallback];
            }
        }
    }
}

插入排序

插入排序是從一個亂序的陣列中依次取值,插入到一個已經排好序的陣列中。
這看起來好像要兩個陣列才能完成,但如果只想在同一個陣列內排序,也是可以的。此時需要想象出兩個區域:前方有序區和後方亂序區。

  1. 分割槽。開始時前方有序區只有一個元素,就是陣列的第一個元素。然後把從第二個元素開始直到結尾的陣列作為亂序區。
  2. 從亂序區取第一個元素,把它正確插入到前方有序區中。把它與前方無序區的最後一個元素比較,亦即與它的前一個元素比較。
    • 如果比前一個元素要大,則不需要交換,這時有序區擴充一格,亂序區往後縮減一格,相當於直接拼在有序區末尾。
    • 如果和前一個元素相等,則繼續和前二元素比較、再和前三元素比較......如果往前遍歷到頭了,發現前方所有元素值都長一個樣的話(囧),那也可以,不需要交換,這時有序區擴充一格,亂序區往後縮減一格,相當於直接拼在有序區末尾。如果比前一個元素大呢?對不起作為有序區不可能出現這種情況。如果比前一個元素小呢,請看下一點。
    • 如果比前一個元素小,則交換它們的位置。交換完後,繼續比較取出元素和它此時的前一個元素,若更小就交換,若相等就比較前一個,直到遍歷完成。
      至此,把亂序區第一個元素正確插入到前方有序區中。
  3. 往後縮小亂序區範圍,繼續取縮小範圍後的第一個元素,重複第2步驟。直到範圍不能縮小為止,排序完成。

插入排序.gif
- (void)jx_insertionSortUsingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    if (self.count == 0) {
        return;
    }
    for (NSInteger i = 1; i < self.count; i ++) {
        for (NSInteger j = i; j > 0 && comparator(self[j], self[j - 1]) == NSOrderedAscending; j --) {
            [self jx_exchangeWithIndexA:j indexB:j - 1 didExchange:exchangeCallback];
        }
    }
}

快速排序

快排的版本有好幾種,粗略可分為:

  • 原始的快排。
  • 為製造適合高效排序環境而事先打亂陣列順序的快排。
  • 為陣列內大量重複值而優化的三向切分快排。

這裡只討論原始的快排。
關於在快排過程中何時進行交換以及交換誰的問題,我看見兩種不同的思路:

  1. 當左右兩個遊標都停止時,交換兩個遊標所指向元素。樞軸所在位置暫時不變,直到兩個遊標相遇重合,才更新樞軸位置,交換樞軸與遊標所指元素。
  2. 當右遊標找到一個比樞軸小的元素時,馬上把樞軸交換到遊標所在位置,而遊標位置的元素則移到樞軸那裡。完成一次樞軸更新。然後左遊標再去尋找比樞軸大的元素,同理。

第1種思路可以有效降低交換頻率,在遊標相遇後再對樞軸進行定位,這步會導致略微增加了比較的次數;
第2種思路交換操作會比較頻繁,但是在交換的過程中同時也把樞軸的位置不斷進行更新,當遊標相遇時,樞軸的定位也完成了。
在兩種思路都嘗試實現過後,我還是喜歡第2種,即便交換操作會多一些,但實質上的交換隻是對陣列特定位置的賦值,這種操作還是挺快的。

  1. 從待排序陣列中選一個值作為分割槽的參考界線,一般選第一個元素即可。這個選出來的值可叫做樞軸pivot,它將會在一趟排序中不斷被移動位置,只終移動到位於整個陣列的正確位置上。
  2. 一趟排序的目標是把小於樞軸的元素放在前方,把大於樞軸的元素放在後方,樞軸放在中間。這看起來一趟排序實質上所幹的事情就是把陣列分割槽。接下來考慮怎麼完成一次分割槽。
  3. 記一個遊標i,指向待排序陣列的首位,它將會不斷向後移動;
    再記一個遊標j,指向待排序陣列的末位,它將會不斷向前移動。
    這樣可以預見的是,ij終有相遇時,當它們相遇的時候,就是這趟排序完成時。
  4. 現在讓遊標j從後往前掃描,尋找比樞軸小的元素x,找到後停下來,準備把這個元素扔到前方去。
  5. 在同一個陣列內排序並不能擴大陣列的容量,那怎麼扔呢?
    因為剛才把首位元素選作為pivot,所以當前它們的位置關係是pivot ... x
    又排序目標是升序,x是個小值卻放在了pivot的後方,不妥,需要交換它們的位置。
  6. 交換完後,它們的位置關係變成了x ... pivot。此時j指向了pivoti指向了x
  7. 現在讓遊標i向後掃描,尋找比樞軸大的元素y,找到後停下來,與pivot進行交換。
    完成後的位置關係是pivot ... y,此時i指向pivot,即pivot移到了i的位置。
  8. 這裡有個小優化,在i向後掃描開始時,i是指向x的,而在上一輪j遊標的掃描中我們已經知道x是比pivot小的,所以完全可以讓i跳過x,不需要拿著xpivot再比較一次。
    結論是在j遊標的交換完成後,順便把i往後移一位,i ++
    同理,在i遊標的交換完成後,順便把j往前移一位,j --
  9. 在掃描的過程中如果發現與樞軸相等的元素怎麼辦呢?
    因我們不討論三向切分的快排優化演算法,所以這裡答案是:不理它。
    隨著一趟一趟的排序,它們會慢慢被更小的元素往後擠,被更大的元素往前擠,最後的結果就是它們都會和樞軸一起移到了中間位置。
  10. ij相遇時,ij都會指向pivot。在我們的分割槽方法裡,把i返回,即在分割槽完成後把樞軸位置返回。
  11. 接下來,讓分出的兩個陣列分別按上述步驟各自分割槽,這是個遞迴的過程,直到陣列不能再分時,排序完成。

快速排序是很天才的設計,實現不復雜,關鍵是它真的很快~


快速排序.gif
- (void)jx_quickSortUsingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    if (self.count == 0) {
        return;
    }
    [self jx_quickSortWithLowIndex:0 highIndex:self.count - 1 usingComparator:comparator didExchange:exchangeCallback];
}

- (void)jx_quickSortWithLowIndex:(NSInteger)low highIndex:(NSInteger)high usingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    if (low >= high) {
        return;
    }
    NSInteger pivotIndex = [self jx_quickPartitionWithLowIndex:low highIndex:high usingComparator:comparator didExchange:exchangeCallback];
    [self jx_quickSortWithLowIndex:low highIndex:pivotIndex - 1 usingComparator:comparator didExchange:exchangeCallback];
    [self jx_quickSortWithLowIndex:pivotIndex + 1 highIndex:high usingComparator:comparator didExchange:exchangeCallback];
}

- (NSInteger)jx_quickPartitionWithLowIndex:(NSInteger)low highIndex:(NSInteger)high usingComparator:(JXSortComparator)comparator didExchange:(JXSortExchangeCallback)exchangeCallback {
    id pivot = self[low];
    NSInteger i = low;
    NSInteger j = high;

    while (i < j) {
        // 略過大於等於pivot的元素
        while (i < j && comparator(self[j], pivot) != NSOrderedAscending) {
            j --;
        }
        if (i < j) {
            // i、j未相遇,說明找到了小於pivot的元素。交換。
            [self jx_exchangeWithIndexA:i indexB:j didExchange:exchangeCallback];
            i ++;
        }

        /// 略過小於等於pivot的元素
        while (i < j && comparator(self[i], pivot) != NSOrderedDescending) {
            i ++;
        }
        if (i < j) {
            // i、j未相遇,說明找到了大於pivot的元素。交換。
            [self jx_exchangeWithIndexA:i indexB:j didExchange:exchangeCallback];
            j --;
        }
    }
    return i;
}

原始碼

https://github.com/JiongXing/JXSort

參考

Swift演算法俱樂部中文版 -- 插入排序
演算法筆記-排序01:選擇排序,插入排序,希爾排序
演算法筆記-排序02:歸併排序,快速排序
1.2-交換排序-快速排序



文/樑炯幸(簡書作者)
原文連結:http://www.jianshu.com/p/70619984fbc6
著作權歸作者所有,轉載請聯絡作者獲得授權,並標註“簡書作者”。

相關文章