排序就是將一組對象按照某種邏輯順序從新排列的過程。這裏咱們主要關注從新排列含有元素的數組 (arrays of items)的算法,其中每一個元素都有一個主鍵 (key)。排序算法的目的是從新排列全部元素,使得元素的主鍵能以某種方式排列。如下代碼是本章通用的排序算法模板:java
public class Example { public static void sort(Comparable[] a) { /*see text*/ } private static boolean less(Comparable v, Comparable w) { return v.compareTo(w) < 0; } private static void exch(Comparable[] a, int i, int j) { Comparable t = a[i]; a[i] = a[j]; a[j] = t; } private static void show(Comparable[] a) { for (int i = 0; i < a.length; i++) StdOut.print(a[i] + " "); StdOut.println(); } public static boolean isSorted(Comparable[] a) { for (int i = 1; i < a.length; i++) if (less(a[i], a[i-1])) return false; return true; } public static void main(String[] args) { String[] a = In.readStrings(); sort(a); assert isSorted(a); show(a); } }
此接口強行對實現它的每一個類的對象進行總體排序。Java中的封裝數字的類型Integer和Double,以及String和其餘許多高級數據類型都實現了Comparable接口,所以能夠直接用這些類型的數組做爲參數調用排序算法,以下圖快速排序調用數組a:算法
在建立本身的數據類型時,只要實現Comparable接口就能保證用例代碼能夠將其排序。要作到這一點,須要實現一個compareTo ()方法來定義目標類型對象的天然次序 。以下圖的Date數據類型:shell
compareTo() 方法 實現了對主鍵 (key) 的抽象 —— 對於任何實現了Comparable接口的數據類型,該方法定義了對象的大小順序。數組
public class Selection { public static void sort(Comparable[] a) { int N = a.length; for (i = 0; i < N; i++) { int min = i; for (j = i+1; j < N; j++) if (less(a[j], a[min])) min = j; exch(a, i, min); } } }
對於長度爲N的數組,選擇排序須要大約 \(~N^2/2\)次比較和N次交換。數據結構
選擇排序的特色:less
public class Insertion { public static void sort(Comparable[] a) { int N = a.length; for (int i = 1; i < N; i++) { for (int j = i; j > 0 && less(a[j], a[j-1]); j--) exch(a, j, j-1); } } }
和選擇排序不一樣,插入排序所需的時間取決於輸入元素的初始順序,對一個很大且其中元素已經有序 (或接近有序) 的數組進行排序要比對隨機數組或逆序數組進行排序要快得多。
提升插入排序速度的方式:在內循環中將較大的元素向右移動而不老是交換兩個元素(這樣訪問數組的次數就能減半)。dom
public class InsertionX { public static void sort(Comparable[] a) { int N = a.length; for (int i = 1; i < N; i++) { Comparable temp = a[i]; for (j = i; j > 0 && less(temp, a[j-1]); j--) a[j] = a[j-1]; } a[j] = temp; } } // alternative way public class InsertionX { public static void sort(Comparable[] a) { int N = a.length; for (int i = 1; i < N; i++) { Comparable temp = a[i]; int j = i; while (j > 0 && less(temp, a[j-1])) { a[j] = a[j-1]; j--; } a[j] = temp; } } }
public class SortCompare { public static double time(String alg, Comparable[] a) { Stopwatch timer = new Stopwatch(); if (alg.equals("Insertion")) Insertion.sort(a); if (alg.equals("Selection")) Selection.sort(a); if (alg.equals("Shell")) Shell.sort(a); if (alg.equals("Merge")) Merge.sort(a); if (alg.equals("Quick")) Quick.sort(a); if (alg.equals("Heap")) Heap.sort(a); return timer.elapsedTime(); } public static double timeRandomInput(String alg, int N, int T) { double total = 0.0; Double[] a = new Double[N]; for (int t = 0; t < T; t++) { for (int i = 0; i < N; i++) a[i] = StdRandom.uniform(); total += time(alg, a); } return total; } public static void main(String[] args) { String alg1 = args[0]; String alg2 = args[1]; int N = Integer.parseInt(args[2]); int T = Integer.parseInt(args[3]); double t1 = timeRandomInput(alg1, N, T); double t2 = timeRandomInput(alg2, N, T); StdOut.printf("For %d random Doubles\n %s is", N, alg1); StdOut.printf(" %.1f times faster than %s\n", t2/t1, alg2); } } /********************************************** % java SortCompare Insertion Selection 1000 100 For 1000 random Doubles Insertion is 1.7 times faster than Selection **********************************************/
希爾排序是插入排序的改進,插入排序的缺點是每次只能交換相鄰的元素,若主鍵最小的元素剛好在數組的末尾,則須要N-1次移動。希爾排序的思想是經過交換不相鄰的元素使數組中任意間隔爲h的元素都是有序的,並最終用插入排序將局部有序的數組排序。ide
public class Shell { public static void sort(Comparable[] a) { int N = a.length; int h = 1; while (h < N/3) h = 3*h + 1; while (h >= 1) { for (int i = h; i < N; i++) { for (int j = i; j >= h && less(a[j], a[j-h]); j -= h) exch(a, j, j-h) } h = h/3; } } } /******************************************** % java SortCompare Shell Insertion 100000 100 For 100000 random Doubles Shell is 600 times faster than Insertion *********************************************/
歸併排序通常是 (遞歸地) 將數組分紅兩半分別排序,而後將結果歸併起來。歸併排序的優勢是所需的時間和 \(NlgN\) 成正比,這使得其優於選擇排序和插入排序。缺點是輔助數組所需的額外空間與N的大小成正比。歸併排序是分治 (divide-and-conquer) 思想的典型體現。性能
public class Merge { private static Comparable[] aux; public static void sort(Comparable[] a) { aux = new Comparable[a.length]; sort(a, 0, a.length - 1); } private static void sort(Comparable[] a, int lo, int hi) { if (hi <= lo) return; int mid = lo + (hi - lo) / 2; sort(a, lo, mid); sort(a, mid+1, hi); merge(a, lo, mid, hi); } public static boid merge(Comparable[] a, int lo, int mid, int hi) { int i = lo, j = mid+1; for (int k = lo; k <= hi; k++) aux[k] = a[k]; for (int k = lo; k <= hi; k++) if (i > mid) a[k] = aux[j++]; else if (j > hi) a[k] = aux[i++]; else if (less(aux[j], aux[i])) a[k] = aux[j++]; else a[k] = aux[i++]; } private static boolean less(Comparable v, Comparable w) { return v.compareTo(w) < 0; } }
public class MergeBU { private static Comparable[] aux; public static void sort(Comparable[] a) { int N = a.length; aux = new Cmparable[N]; for ( int sz = 1; sz < N; sz = sz+sz) for (int lo = 0; lo < N-sz; lo += sz+sz) merge(a, lo, lo+sz-1, Math.min(lo+sz+sz-1, N-1)); } public staic boid merge(Comparable[] a, int lo, int mid, int hi) { int i = lo, j = mid+1; for (int k = lo; k <= hi; k++) aux[k] = a[k]; for (int k = lo; k <= hi; k++) if (i > mid) a[k] = aux[j++]; else if (j > hi) a[k] = aux[i++]; else if (less(aux[j], aux[i])) a[k] = aux[j++]; else a[k] = aux[i++]; } private static boolean less(Comparable v, Comparable w) { return v.compareTo(w) < 0; } }
在通常應用中快速排序都比其餘排序算法快得多,除此以外快速排序是原地排序,所需時間和 \(NlgN\)成正比。主要缺點是很是脆弱,在實現時須要很是當心才能避免低劣的性能。測試
快速排序也是一種分治的排序算法。與歸併排序的區別:歸併排序將兩個子數組分別排序後,歸併前整個數組不是有序的;快速排序則是當子數組都有序時整個數組也就天然有序了。在歸併排序中,一個數組被分紅兩半;快速排序中,切分 (partition) 的位置取決於數組的內容。
快速排序遞歸地將子數組 a[lo...hi] 排序,先用partition()方法將a[j] 放到一個合適位置,而後遞歸地將其餘位置的元素排序。這裏的關鍵在於切分,切分後數組知足三個條件:
切分的實現方法:先取a[lo] 做爲切分元素,而後從數組左端開始向右掃描直到找到一個大於等於它的元素,再從右端開始向左掃描知道找到一個小於等於它的元素,最後交換這兩個元素的位置。當兩個指針相遇時,將切分元素a[lo] 和左子數組最右側的元素 (a[j]) 交換,這樣切分值就留在a[j]中。
public class Quick { public static void sort(Comparable[] a) { StdRandom.shuffle(a); sort(a, 0, a.length - 1); } private static void sort(Comparable[] a, int lo, int hi) { if (hi <= lo) return; int j = partition(a, lo, hi); sort(a, lo, j-1); sort(a, j+1, hi); } private static int partition(Comparable[] a, int lo, int hi) { int i = lo, j = hi + 1; Comparable v = a[lo]; while (true) { while (less(a[++i], v)) if (i == hi) break; while (less(v, a[--j])) if (j == lo) break; if (i >= j) break; exch(a, i, j); } exch(a, lo, j); return j; } }
上述快速排序的實現有一個潛在的缺點:在切分不平衡時可能會極爲低效,達到平方級別。例如若是從最小的元素切分,第二次從第二小的元素切分,如此這般每次只移除一個元素,這會致使一個大數組須要切分不少次。所以快速排序前須要現將數組隨機排序。
切換到插入排序: 對於小數組,快速排序比插入排序慢,因此可將上述算法中sort() 方法從if (hi <= lo) return;
改成 if (hi <= lo + M) { Insertion.sort(a, lo, hi); return; }
當數組中存在大量重複元素時,快速排序的效率能夠經過三向切分排序進一步改進,能將排序時間從線性對數級降低到線性級別。將數組分爲三部分,分別對應小於、等於和大於切分元素的數組元素。使用Comparable 接口對a[i] 進行三向比較:
public class Quick3way { private static void sort(Comparable[] a, int lo, int hi) { if (hi <= lo) return; int lt = lo, i = lo + 1, gt = hi; Comparable v = a[lo]; while (i <= gt) { int cmp = a[i].compareTo(v); if (cmp < 0) exch(a, lt++, i++); else if (cmp > 0) exch(a, i, gt--); else i++; } sort(a, lo, lt - 1); sort(a, gt + 1, hi); } }
普通的隊列 (queue) 是一種先進先出的數據結構,元素在隊列尾追加,而從隊列頭刪除。在優先隊列中,元素被賦予優先級。當訪問元素時,具備最高優先級的元素最早刪除。這種隊列不是直接將新元素放在隊列尾部,而是放在比它優先級低的元素前面。
優先隊列最重要的操做是刪除最大元素 (delMax()) 和插入元素 (insert()) ,見如下API:
如下是一個優先隊列的測試用例,打印數字最大的M行:
public class TopM { public static void main(String[] args) { int M = Integer.parseInt(args[0]); MinPQ<Transaction> pq = new MinPQ<Transaction>(M+1); while (StdIn.hasNextLine()) { pq.insert(new Transaction(StdIn.readLine())); if (pq.size() > M) pd.delMin(); } Stack<Transaction> stack = new Stack<Transaction>(); while (!pq.isEmpty()) stack.push(pq.delMin()); for (Transaction t : stack) StdOut.println(t); } }
數據結構二叉堆 (binary heap) 能很好地實現優先隊列的基本操做。二叉堆是一組可以用堆有序的徹底二叉樹排序的元素,並在數組中按照層級存儲 (不使用數組的第一個位置)。堆有序 (heap-ordered) 指的是一顆二叉樹的每一個結點都大於等於它的兩個子結點。
在一個堆中,位置爲k的結點的父結點位置爲\(\left\lfloor k/2 \right\rfloor\) ,兩個子結點的位置分別爲 \(2k\) 和 \(2k+1\)。一顆大小爲N的徹底二叉樹的高度爲\(\left\lfloor lgN \right\rfloor\)
自下而上的堆有序化 —— 上浮 (swim)
若是某個結點比其父結點大,則須要交換兩者以保持堆的有序化,這樣的循環過程爲swim() 方法:
private void swim(int k) { while (k > 1 && less(k/2, k)) { exch(k/2, k); k = k/2; } }
自上而下的堆有序化 —— 下沉 (sink)
若是某個結點比其兩個子結點的其中之一小,則須要將它和兩個子結點的較大者交換以保持堆的有序化這樣的循環過程爲sink() 方法:
private void sink(int k) { while (2*k <= N) { int j = 2*k; if (j < N && less(j, j+1)) j++; if (!less(k, j)) break; exch(k, j); k = j; } }
插入元素:將新元素加到數組末尾,增長堆的大小並將這個元素上浮到合適位置。
刪除最大元素:從數組頂端刪去最大元素並將數組最後一個元素放到頂端,再讓這個元素下沉到合適位置。
下列代碼實現了優先隊列,優先隊列由一個基於堆的徹底二叉樹表示,存儲於數組 pq[1...N] 中,pq[0] 沒有使用。
public class MaxPQ<Key extends Comparable<Key>> { private Key[] pq; private int N = 0; public MaxPQ(int maxN) { pq = (Key[]) new Comparable[maxN+1]; } public boolean isEmpty() { return N == 0; } public int size() { return N; } public void insert(Key v) { pq[++N] = v; swim(N); } public Key delMax() { Key max = pq[1]; exch(1, N--); pq[N+1] = null; sink(1); return max; } private boolean less(int i, int j) { return pq[i].compareTo(pq[j]) < 0; } private void exch(int i, int j) { Key t = pq[i]; pq[i] = pq[j]; pq[j] = t; } private void swim(int k) { while (k > 1 && less(k/2, k)) { exch(k/2, k); k = k/2; } } private void sink(int k) { while (2*k <= N) { int j = 2*k; if (j < N && less(j, j+1)) j++; if (!less(k, j)) break; exch(k, j); k = j; } } }
索引優先隊列即在優先隊列的基礎上給數組中每一個元素添加一個索引。如下實現用pq[]
保存索引,qp[]
保存pq[]
的逆序,即pq[]
中元素的位置。舉例:pq = [1,2,0], 則qp = [2,0,1],pq[qp[1]]=1
。數組keys[]
保存元素。
import edu.princeton.cs.algs4.StdOut; import edu.princeton.cs.algs4.In; public class IndexMinPQ<Key extends Comparable<Key>> { private int maxN; private int n; private int[] pq; private int[] qp; private Key[] keys; public IndexMinPQ(int maxN) { this.maxN = maxN; n = 0; keys = (Key[]) new Comparable[maxN + 1]; pq = new int[maxN + 1]; qp = new int[maxN + 1]; for (int i = 0; i <= maxN; i++) qp[i] = -1; } public boolean isEmpty() { return n == 0; } public boolean contains(int i) { return qp[i] != -1; } public int size() { return n; } public void insert(int i, Key key) { n++; qp[i] = n; pq[n] = i; keys[i] = key; swim(n); } public int minIndex() { return pq[1]; } public Key minKey() { return keys[pq[1]]; } public int delMin() { int min = pq[1]; exch(1, n--); sink(1); assert min == pq[n+1]; qp[min] = -1; keys[min] = null; pq[n+1] = -1; return min; } public void changeKey(int i, Key key) { keys[i] = key; swim(qp[i]); sink(qp[i]); } public void delete(int i) { int index = qp[i]; exch(index, n--); swim(index); sink(index); keys[i] = null; qp[i] = -1; } private boolean greater(int i, int j) { return keys[pq[i]].compareTo(keys[pq[j]]) > 0; } private void exch(int i, int j) { int swap = pq[i]; pq[i] = pq[j]; pq[j] = swap; qp[pq[i]] = i; qp[pq[j]] = j; } private void swim(int k) { while (k > 1 && greater(k/2, k)) { exch(k, k/2); k = k/2; } } private void sink(int k) { while (2*k <= n) { int j = 2*k; if (j < n && greater(j, j+1)) j++; if (!greater(k, j)) break; exch(k, j); k = j; } } /****************************************************** 如下爲用例:多向歸併問題,將多個有序的輸入流歸併成一個有序的輸出流 ******************************************************/ public static void merge(In[] streams) { int N = streams.length; IndexMinPQ<String> pq = new IndexMinPQ<String>(N); for (int i = 0; i < N; i++) if (!streams[i].isEmpty()) pq.insert(i, streams[i].readString()); while (!pq.isEmpty()) { StdOut.printf(pq.minKey() + " "); int i = pq.delMin(); if (!streams[i].isEmpty()) pq.insert(i, streams[i].readString()); } } public static void main(String[] args) { int N = args.length; In[] streams = new In[N]; for (int i = 0; i < N; i++) streams[i] = new In(args[i]); merge(streams); StdOut.println(); } } /************* 輸入: % more m1.txt A B C F G I I Z % more m2.txt B D H P Q Q % more m3.txt A B E F J 輸出: % java IndexMinPQ m1.txt m2.txt m3.txt A A B B B C D E F F G H I I J N P Q Q Z **************************************/
堆排序可分爲兩個階段:
堆的構造經過從下到上遞歸地構建子堆來實現。若是一個結點的兩個子結點都已是堆了,那麼在該結點上調用sink() 就能把它們變成一個堆,這樣就能按數組逆序從下而上地構建堆。只需對數組中一半元素使用 sink()方法,由於剩下的元素都是葉結點。完成後堆的最大元素將位於數組的開頭。
下沉排序階段將頂端和末端元素進行交換,而後將最大元素從堆中刪除並用 sink() 從新調整堆的結構,如此循環最後獲得一個有序序列。
import edu.princeton.cs.algs4.StdIn; import edu.princeton.cs.algs4.StdOut; public class Heap { private Heap() { } public static void sort(Comparable[] pq) { int n = pq.length; for (int k = n/2; k >= 1; k--) sink(pq, k, n); while (n > 1) { exch(pq, 1, n--); sink(pq, 1, n); } } private static void sink(Comparable[] pq, int k, int n) { while (2*k <= n) { int j = 2*k; if (j < n && less(pq, j, j+1)) j++; if (!less(pq, k, j)) break; exch(pq, k, j); k = j; } } private static boolean less(Comparable[] pq, int i, int j) { return pq[i-1].compareTo(pq[j-1]) < 0; } private static void exch(Object[] pq, int i, int j) { Object swap = pq[i-1]; pq[i-1] = pq[j-1]; pq[j-1] = swap; } private static void show(Comparable[] a) { for (int i = 0; i < a.length; i++) StdOut.println(a[i]); } public static void main(String[] args) { String[] a = StdIn.readAllStrings(); Heap.sort(a); show(a); } }
只需線性對數的時間,先將數組排序,再遍歷數組,記錄連續出現的重複元素。若是用平方級別的算法則要將全部元素相互比較一遍。
Quick.sort(a); int count = 1; for (int i = 1; i < a.length; i++) if (a[i].compareTo(a[i-1]) != 0) count++;
/