數(shù)據(jù)結(jié)構(gòu)與算法(五)校翔,優(yōu)先隊列

這節(jié)總結(jié)一下優(yōu)先隊列的常用實現(xiàn)方法弟跑。

目錄:

  • 1、基本概念

  • 2防症、基于數(shù)組實現(xiàn)的優(yōu)先隊列

  • 2.1孟辑、基于有序數(shù)組的實現(xiàn)

  • 2.2、基于無序數(shù)組的實現(xiàn)

  • 3蔫敲、基于堆實現(xiàn)的優(yōu)先隊列

  • 3.1饲嗽、堆的有序化

  • 3.2、基于堆實現(xiàn)的優(yōu)先隊列

  • 4奈嘿、索引優(yōu)先隊列

1貌虾、基本概念

普通的隊列是一種先進先出的數(shù)據(jù)結(jié)構(gòu),元素在隊列尾追加裙犹,而從隊列頭刪除尽狠。在優(yōu)先隊列中衔憨,元素被賦予優(yōu)先級。當(dāng)訪問元素時袄膏,具有最高優(yōu)先級的元素最先刪除践图。優(yōu)先隊列具有最高級先出 (largest-in,first-out)的行為特征沉馆。(百度百科)

抽象數(shù)據(jù)類型:

優(yōu)先隊列的接口同前面講到的隊列的接口一樣码党,是其基于泛型的API接口代碼如下:

public interface Queue<E> {

    //隊列是否為空
    boolean isEmpty();

    //隊列的大小
    int size();

    //入隊
    void enQueue(E element);

    //出隊
    E deQueue();
}

2、基于數(shù)組實現(xiàn)的優(yōu)先隊列

實現(xiàn)優(yōu)先隊列最簡的方法就是基于前面講到的基于數(shù)組的棧的代碼斥黑,只需對插入或刪除操作作相應(yīng)的更改即可揖盘。

2.1、基于有序數(shù)組的實現(xiàn)

在棧的代碼的插入方法中添加代碼心赶,將所有較大的元素向右移動一格扣讼,以保證數(shù)組有序(和插入排序相同),這里我們可以使用二分查找的方法來找出元素應(yīng)插入的位置缨叫,然后再移動元素椭符。這樣最大元素,總是在數(shù)組的最右邊耻姥,其刪除操作和棧的實現(xiàn)中一樣销钝。

代碼:

/**
 * 基于有序數(shù)組的實現(xiàn)的優(yōu)先隊列
 * @author Alent
 * @param <E>
 */
public class PriorityQueue<E extends Comparable<E>> implements Queue<E>{
    private E[] elements;
    private int size=0;
    
    @SuppressWarnings("unchecked")
    public PriorityQueue() {
        elements = (E[])new Comparable[1]; 
    }
    
    @Override public int size() {return size;}

    @Override public boolean isEmpty() {return size == 0;}

    @Override
    public void enQueue(E element) {
        if(size == elements.length) {
            resizingArray(2*size);//若數(shù)組已滿將長度加倍
        }
        elements[size++] = element;
        insertSort(elements);
    }

    @Override
    public E deQueue() {
        E element = elements[--size];
        elements[size] = null;     //注意:避免對象游離
        if(size > 0 && size == elements.length/4) {
            resizingArray(elements.length/2);//小于數(shù)組1/4,將數(shù)組減半
        }
        return element;
    }
    
    //插入排序,由于前面n-1個元素是有序的琐簇,這里只插入最后一個元素
    public void insertSort(E[] a) {
        int N = size -1; //最后一個元素是size-1蒸健,不是a.length-1
        if(N == 0) return;
        int num = binaryFind(a, a[N], 0, N-1);
        E temp = a[N];
        //num后的元素向后移動
        for (int j = N; j > num; j--) {
           a[j] = a[j-1];
        }
        a[num] = temp;
    }

    //找出元素應(yīng)在數(shù)組中插入的位置
    public int binaryFind(E[] a, E temp, int down, int up) {
        if(up<down || up>a.length || down<0) {
            System.out.println("下標錯誤");
        }
        if(temp.compareTo(a[down]) < 0) return down;
        if(temp.compareTo(a[up]) > 0) return up+1;
        int mid = (up-down)/2 + down;
        if(temp.compareTo(a[mid]) == 0) {
            return mid + 1;
        }else if(temp.compareTo(a[mid])<0) {
            up = mid-1;
        }else if(temp.compareTo(a[mid])>0) {
            down = mid+1;
        }
        return binaryFind(a,temp,down,up);
    }

    //交換兩個元素
    public void swap(E[] a,int i,int j) {
        E temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }
    
    //調(diào)整數(shù)組大小
    public void resizingArray(int num) {
        @SuppressWarnings("unchecked")
        E[] temp = (E[])new Comparable[num];
        for(int i=0;i<size;i++) {
            temp[i] = elements[i];
        }
        elements = temp;
    }
    
    public static void main(String[] args) {
        int[] a = {4,2,1,3,8,new Integer(5),7,6};//測試數(shù)組
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
        System.out.print("入棧順序:");
        for(int i=0;i<a.length;i++) {
            System.out.print(a[i]+" ");
            pq.enQueue(a[i]);
        }
        System.out.println();
        System.out.print("出棧順序數(shù)組實現(xiàn):");
        while(!pq.isEmpty()) {
            System.out.println(pq.deQueue());
        }
    }
}

2.2、基于無序數(shù)組的實現(xiàn)

同樣婉商,我們一個可以在刪除方法中修改似忧,在刪除方法中添加一段類似于選擇排序內(nèi)循環(huán)的代碼,每次刪除時先找出數(shù)組中的最大元素丈秩,然后與最右邊元素進行交換盯捌,然后在刪除元素。

代碼:

@Override
public void enQueue(E element) {
    if(size == elements.length) {
        resizingArray(2*size);//若數(shù)組已滿將長度加倍
    }
    elements[size++] = element;
}

@Override
public E deQueue() {
    swapMax(elements);
    E element = elements[--size];
    elements[size] = null;     //注意:避免對象游離
    if(size > 0 && size == elements.length/4) {
        resizingArray(elements.length/2);//小于數(shù)組1/4,將數(shù)組減半
    }
    return element;
}

public void swapMax(E[] a) {
    int max = size -1;
    for(int i=0;i<size-1; i++) {
        if(larger(a[i],a[max])) 
            max = i;
    }
    swap(a, size-1, max);
}

//比較兩個元素大小
public boolean larger(E a1, E a2) {
    return a1.compareTo(a2)>0;
}

3蘑秽、基于堆實現(xiàn)的優(yōu)先隊列

基本概念:

當(dāng)一個二叉樹的每個結(jié)點都大于等于它的兩個子結(jié)點時饺著,我們稱它是堆有序的。根結(jié)點是堆有序的二叉樹的最大結(jié)點肠牲。

二叉堆是一組能夠用堆有序的完全二叉樹排序的元素幼衰,并在數(shù)組中按照層級存儲。

一棵堆有序的完全二叉樹:


堆有序的完全二叉樹

為了操作方便缀雳,這是我們使用一個數(shù)組渡嚣,來表示一個堆。我們不使用數(shù)組的第一個元素,具體實現(xiàn)在《數(shù)據(jù)結(jié)構(gòu)與算法(四)严拒,樹》中有提及扬绪,這里就不說了。

3.1裤唠、堆的有序化

當(dāng)我們將元素插入到堆(數(shù)組的末尾)中時,插入的元素可能比它的父結(jié)點要大莹痢,堆的有序狀態(tài)被打破种蘸。我們需要交換它和它的父節(jié)點來修堆,直到堆重新變?yōu)橛行驙顟B(tài)竞膳。其操作如下圖:

堆的插入操作

代碼如下:

//上浮操作
private void swim(int k) {
        while(k > 1 && less(k/2, k)) {
            swap(k/2, k);
            k = k/2;
        }
    }

private boolean less(int i, int j) {
    return elements[i].compareTo(elements[j]) < 0;
}

//交換兩個元素
public void swap(int i,int j) {
    E temp = elements[i];
    elements[i] = elements[j];
    elements[j] = temp;
}

同樣的航瞭,當(dāng)我們從堆中刪除根結(jié)點并將它的最后一個元素放到頂端時,堆的有序性被打破坦辟,我們需要將它與它的兩個子結(jié)點種的較大者進行交換刊侯,以恢復(fù)堆的有序性,其操作流程如下圖:

堆的刪除操作

其代碼如下:

//下沉操作
private void sink(int k) {
    while(2*k <= size) {
        int j = 2*k;
        if(j < size && less(j, j+1))
            j++;
        if(!less(k,j))
            break;
        swap(k,j);
        k = j;
    }
}

3.2锉走、基于堆實現(xiàn)的優(yōu)先隊列

基于堆的優(yōu)先隊列的實現(xiàn)代碼如下:

/**
 * 基于堆的優(yōu)先隊列
 * @author Alent
 */
public class MaxPQ<E extends Comparable<E>> implements Queue<E>{
    private E[] elements;
    private int size=0;
    
    @SuppressWarnings("unchecked")
    public MaxPQ(int capacity) {
        elements = (E[])new Comparable[capacity + 1]; 
    }
    
    @Override public int size() {return size;}

    @Override public boolean isEmpty() {return size == 0;}

    @Override
    public void enQueue(E element) {
        elements[++size] = element;
        swim(size);
    }
    
    //上浮
    private void swim(int k) {
        while(k > 1 && less(k/2, k)) {
            swap(k/2, k);
            k = k/2;
        }
    }
    
    private boolean less(int i, int j) {
        return elements[i].compareTo(elements[j]) < 0;
    }

    @Override
    public E deQueue() {
        E result = elements[1];
        swap(1, size--);
        elements[size + 1] = null;
        sink(1);
        return result;
    }
    
    //下沉
    private void sink(int k) {
        while(2*k <= size) {
            int j = 2*k;
            if(j < size && less(j, j+1))
                j++;
            if(!less(k,j))
                break;
            swap(k,j);
            k = j;
        }
    }

    //交換兩個元素
    public void swap(int i,int j) {
        E temp = elements[i];
        elements[i] = elements[j];
        elements[j] = temp;
    }
}

三種實現(xiàn)方法的時間復(fù)雜度比較:

4滨彻、索引優(yōu)先隊列

索引優(yōu)先隊列,它用一個索引數(shù)組保存了某個元素在優(yōu)先隊列中的位置挪蹭,使得我們能夠引用已經(jīng)進入優(yōu)先隊列中的元素亭饵。最在些應(yīng)用中,通常是很有必要的梁厉,如:有向圖的Dijkstra算法中就使用了索引優(yōu)先隊列辜羊,來返回最小邊的索引。

其實現(xiàn)方法為:

使用elements[]數(shù)組來保存隊列中的元素词顾,pq[]數(shù)組用來保存elements中元素的索引八秃,在添加一個數(shù)組qp[]來保存pq[]的逆序——qp[i]的值是i在pq[]中的位置(即 pq[qp[i]] = i)。若i不在隊列中肉盹,則令qp[i] = -1昔驱。輔助函數(shù)less()、swap()垮媒、sink()舍悯、swim()和前面優(yōu)先隊列中的一樣。

索引優(yōu)先隊列的代碼實現(xiàn):

/**
 * 基于堆實現(xiàn)的索引優(yōu)先隊列
 */
public class IndexMinPQ<E extends Comparable<E>>{

    private int[] pq; //索引二叉堆
    private int[] qp; // 保存逆序:pq[qp[i]] = i;
    private E[] elements; //元素
    private int size = 0;

    @SuppressWarnings("unchecked")
    public IndexMinPQ(int capacity) {
        elements = (E[]) new Comparable[capacity + 1];
        pq = new int[capacity + 1];
        qp = new int[capacity + 1];
        for (int i = 0; i <= capacity; i++) {
            qp[i] = -1;
        }
    }

    public boolean isEmpty() {
        return size == 0;
    }
    //刪除最小元素睡雇,并返回索引
    public int delMin() {
        int index = pq[1];
        swap(1, size--);
        sink(1);
        elements[pq[size + 1]] = null;
        qp[pq[size + 1]] = -1;
        return index;
    }
    //刪除索引k及其元素
    public void delete(int k) {
        int index = qp[k];
        swap(index, size--);
        swim(index);
        sink(index);
        elements[k] = null;
        qp[k] = -1;
    }
    
    //插入元素萌衬,將它和索引k關(guān)聯(lián)
    public void insert(int k, E element) {
        size++;
        qp[k] = size;
        pq[size] = k;
        elements[k] = element;
        swim(size);
    }

     //改變索引k關(guān)聯(lián)的元素
    public void change(int k, E element) {
        elements[k] = element;
        swim(qp[k]);
        sink(qp[k]);
    }
    
    //是否包含索引k
    public boolean contains(int k) {
        return qp[k] != -1;
    }
    
    //下沉
    private void sink(int k) {
        while (2 * k <= size) {
            int j = 2 * k;
            if (j < size && less(j, j + 1))
                j++;
            if (!less(k, j))
                break;
            swap(k, j);
            k = j;
        }
    }

    //上浮
    private void swim(int k) {
        while (k > 1 && less(k / 2, k)) {
            swap(k, k / 2);
            k = k / 2;
        }
    }
    private boolean less(int i, int j) {
        return elements[pq[i]].compareTo(elements[pq[j]]) > 0;
    }

    //交換兩元素
    private void swap(int i, int j) {
        int swap = pq[i];
        pq[i] = pq[j];
        pq[j] = swap;
        qp[pq[i]] = i;
        qp[pq[j]] = j;
    }
}

索引優(yōu)先隊列的時間復(fù)雜度:

索引優(yōu)先隊列的時間復(fù)雜度

找出最大元素的索引優(yōu)先隊列的JAVA版本IndexMaxPQ.java 點這里

好了它抱,這節(jié)就總結(jié)這么多吧秕豫。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子混移,更是在濱河造成了極大的恐慌祠墅,老刑警劉巖,帶你破解...
    沈念sama閱讀 221,820評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件歌径,死亡現(xiàn)場離奇詭異毁嗦,居然都是意外死亡,警方通過查閱死者的電腦和手機回铛,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,648評論 3 399
  • 文/潘曉璐 我一進店門狗准,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人茵肃,你說我怎么就攤上這事腔长。” “怎么了验残?”我有些...
    開封第一講書人閱讀 168,324評論 0 360
  • 文/不壞的土叔 我叫張陵捞附,是天一觀的道長。 經(jīng)常有香客問我您没,道長鸟召,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 59,714評論 1 297
  • 正文 為了忘掉前任紊婉,我火速辦了婚禮药版,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘喻犁。我一直安慰自己槽片,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 68,724評論 6 397
  • 文/花漫 我一把揭開白布肢础。 她就那樣靜靜地躺著还栓,像睡著了一般。 火紅的嫁衣襯著肌膚如雪传轰。 梳的紋絲不亂的頭發(fā)上剩盒,一...
    開封第一講書人閱讀 52,328評論 1 310
  • 那天,我揣著相機與錄音慨蛙,去河邊找鬼辽聊。 笑死,一個胖子當(dāng)著我的面吹牛期贫,可吹牛的內(nèi)容都是我干的跟匆。 我是一名探鬼主播,決...
    沈念sama閱讀 40,897評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼通砍,長吁一口氣:“原來是場噩夢啊……” “哼玛臂!你這毒婦竟也來了烤蜕?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,804評論 0 276
  • 序言:老撾萬榮一對情侶失蹤迹冤,失蹤者是張志新(化名)和其女友劉穎讽营,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體泡徙,經(jīng)...
    沈念sama閱讀 46,345評論 1 318
  • 正文 獨居荒郊野嶺守林人離奇死亡橱鹏,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,431評論 3 340
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了堪藐。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片蚀瘸。...
    茶點故事閱讀 40,561評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖庶橱,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情贪惹,我是刑警寧澤苏章,帶...
    沈念sama閱讀 36,238評論 5 350
  • 正文 年R本政府宣布,位于F島的核電站奏瞬,受9級特大地震影響枫绅,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜硼端,卻給世界環(huán)境...
    茶點故事閱讀 41,928評論 3 334
  • 文/蒙蒙 一并淋、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧珍昨,春花似錦县耽、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,417評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至兄春,卻和暖如春澎剥,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背赶舆。 一陣腳步聲響...
    開封第一講書人閱讀 33,528評論 1 272
  • 我被黑心中介騙來泰國打工哑姚, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人芜茵。 一個月前我還...
    沈念sama閱讀 48,983評論 3 376
  • 正文 我出身青樓叙量,卻偏偏與公主長得像,于是被迫代替她去往敵國和親夕晓。 傳聞我的和親對象是個殘疾皇子宛乃,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,573評論 2 359

推薦閱讀更多精彩內(nèi)容