最小生成樹

前言

在電子電路設計中,我們常常需要將多個組件的針腳連接在一起噪裕,要連接n個針腳株婴,我們可以使用n-1根連線怎虫。很顯然暑认,我們希望所使用的連線最短。

可以將上述布線問題用一個連通無向圖 G = {V, E} 來表示大审,V是針腳的集合蘸际,E是針腳之間的可能連接,并且為每條邊 {u , v}屬于E徒扶,我們?yōu)槠滟x權重 w 作為連接針腳u和v的代價粮彤。我們希望找到一個無環(huán)子集T,既能連接所有的針腳姜骡,又有最小代價导坟。

因為T是無環(huán)子集,并且連通所有結點圈澈,因此T必然是一棵樹惫周,我們稱這樣的樹是圖的最小生成樹。

求解最小生成樹有兩種方法康栈,Kruskal算法和Prim算法闯两。

本文中使用的圖

Kruskal算法

Kruskal算法,在所有連接森林中兩棵不同樹的邊里邊谅将,找到權重最小的邊加入到森林漾狼。

簡單地說,Kruskal算法將邊排序饥臂,根據貪心算法思維逊躁,選擇最短邊。但最小生成樹的結果要求是無環(huán)的隅熙,且每個頂點都需要包括稽煤,如何確保這個結果呢?

將圖中的每個頂點當成一棵單獨的樹囚戚,如果選擇的最短邊為{u酵熙、v},森林中這兩棵樹并未交集驰坊,則合并這兩棵樹到森林中匾二。直到森林中包含所有的結點。

算法導論上的偽碼如下:

Kruskal算法中的難點就在于確保無環(huán)且包含所有結點拳芙,并且是連通圖察藐。

將每個結點都看成一棵樹,如果選中的邊兩頂點位于不同的樹中舟扎,則合并這兩棵樹加入森林分飞,這就間接使兩個頂點相連了。如果兩個已經連通的頂點已經被加入森林中睹限,那么就不符合條件而不會被重復加入譬猫,因此也會是無環(huán)的讯檐。

/*
 * 將邊按權重從小到大排列
 * 如果邊的兩個頂點在不同樹中則將這兩棵樹合并,且加入森林
 * 在森林中的頂點則是連通的染服,已連通的頂點在森林中裂垦,也不會重復添加邊,達到無環(huán)要求
 */
public void kruskal(){
    MatrixGraph graph = initUndigraph();
    MatrixEdge[] array = getEdges(graph, null);
    //為邊排序
    sort(array, 0, array.length-1);
    //構造森林肌索,森林中的元素就是一棵棵樹蕉拢,初始樹中只包含一個頂點
    ArrayList<HashSet<Vertex>> list = new ArrayList<>();
    for (int i = 0; i < graph.mList.size(); i++) {
        HashSet<Vertex> set = new HashSet<>();
        set.add(graph.mList.get(i));
        list.add(set);
    }
    MatrixEdge edge = null;
    for (int i = 0; i < array.length; i++) {
        edge = array[i];
        Vertex v1 = edge.v1;
        Vertex v2 = edge.v2;
        int count1 = -1;
        int count2 = -1;
        for (int j = 0; j < list.size(); j++) {
            HashSet<Vertex> set = list.get(j);
            //找到頂點在森林中的位置
            if (set.contains(v1)) {
                count1 = j;
            }
            if (set.contains(v2)) {
                count2 = j;
            }
        }
        if (count1 == -1 || count2 == -1) {
            return;
        }else {
            if (count1 != count2) {
                //如果頂點分別在不同的樹中,則合并這兩棵樹并且刪除之前的舊樹
                //因為會合并樹诚亚,所以已經連通的頂點就會在合并的樹中晕换,在同一棵樹中
                //count1等于count2,所以能保證無環(huán)
                //最終森林中只有一棵樹站宗,且包含全部頂點
                HashSet<Vertex> set1 = list.get(count1);
                HashSet<Vertex> set2 = list.get(count2);
                set1.addAll(set2);
                if (count1 < count2) {
                    list.remove(count1);
                    list.remove(count2-1);
                }else {
                    list.remove(count2);
                    list.remove(count1-1);
                }
                list.add(set1);
                System.out.println(edge);
            }
        }
    }
}

值得一提的是闸准,圖中兩種存儲方式,矩陣和鏈表梢灭,Kruskal算法需要對所有邊排序夷家,而鄰接鏈表中直觀獲取所有邊比較難,所以需要用鄰接矩陣存儲圖敏释。

Prim算法

Prim算法库快,從圖中任意挑選出一個頂點放入集合list中,在list的所有頂點中選出邊最短的一條邊钥顽,并且將邊的另一頂點也加入list中义屏,如此重復到所有頂點都已加入到list中。

算法步驟:

  • 輸入:一個加權連通圖蜂大,其中頂點集合為V闽铐,邊集合為E
  • 初始化:Vnew = {x},其中x為集合V中的任一節(jié)點(起始點)奶浦,Enew = {},為空兄墅;
  • 重復下列操作,直到Vnew = V:
    a: 在集合E中選取權值最小的邊<u, v>澳叉,其中u為集合Vnew中的元素隙咸,而v不在Vnew集合當中,并且v∈V(如果存在有多條滿足前述條件即具有相同權值的邊耳高,則可任意選取其中之一)扎瓶;
    b: 將v加入集合Vnew中所踊,將<u, v>邊加入集合Enew中
  • 輸出:使用集合Vnew和Enew來描述所得到的最小生成樹泌枪。

prim算法是比kruskal更簡單一點。因為prim算法需要計算以某頂點為起點的最小邊秕岛,所以使用鄰接鏈表存儲圖更合適碌燕。

/*
 * 構造新結點list误证,先從圖中選中任意結點加入
 * 不斷遍歷list中的結點,選擇與結點相關的最短邊修壕,且將邊的另一端加入到list中
 * 選擇的最短邊的另一個結點之前不能在list中
 * 最后list與圖中頂點表相同則結束整個過程
 */
public void prim(){
    Graph graph = initUndigraph();
    Vertex start = graph.mList.get(0);
    //構建新list
    List<Vertex> oldList = graph.mList;
    List<Vertex> newList = new ArrayList<Vertex>();
    newList.add(start);
    Vertex v = null;
    Arc vArc = null;
    //最短邊的權重愈捅、對應的弧、對應的弧起點
    int vMinW = Integer.MAX_VALUE;
    Arc vMinArc = null;
    Vertex vMinVertex = null;
    while (newList.size() < oldList.size()) {
        //查找出新list中最短邊
        for (int i = 0; i < newList.size(); i++) {
            v = newList.get(i);
            vArc = v.firstArc;
            while (vArc != null) {
                if (vMinW > vArc.weight && !newList.contains(vArc.vertex)) {
                    vMinW = vArc.weight;
                    vMinArc = vArc;
                    vMinVertex = v;
                }
                vArc = vArc.next;
            }
        }
        //輸出結果
        if (vMinArc != null && vMinArc.vertex != null && !newList.contains(vMinArc.vertex)) {
            System.out.println(vMinVertex.info + "    " + vMinArc.weight + "    " + vMinArc.vertex.info);
            newList.add(vMinArc.vertex);
            vMinW = Integer.MAX_VALUE;
        }
    }
}

所有代碼均上傳至本人的github慈鸠,歡迎訪問蓝谨。從本文來看,圖的不同存儲方式也能帶來不同的便利青团。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末譬巫,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子督笆,更是在濱河造成了極大的恐慌芦昔,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,640評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件娃肿,死亡現場離奇詭異咕缎,居然都是意外死亡,警方通過查閱死者的電腦和手機料扰,發(fā)現死者居然都...
    沈念sama閱讀 93,254評論 3 395
  • 文/潘曉璐 我一進店門凭豪,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人晒杈,你說我怎么就攤上這事墅诡。” “怎么了桐智?”我有些...
    開封第一講書人閱讀 165,011評論 0 355
  • 文/不壞的土叔 我叫張陵末早,是天一觀的道長。 經常有香客問我说庭,道長然磷,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,755評論 1 294
  • 正文 為了忘掉前任刊驴,我火速辦了婚禮姿搜,結果婚禮上,老公的妹妹穿的比我還像新娘捆憎。我一直安慰自己舅柜,他們只是感情好,可當我...
    茶點故事閱讀 67,774評論 6 392
  • 文/花漫 我一把揭開白布躲惰。 她就那樣靜靜地躺著致份,像睡著了一般。 火紅的嫁衣襯著肌膚如雪础拨。 梳的紋絲不亂的頭發(fā)上氮块,一...
    開封第一講書人閱讀 51,610評論 1 305
  • 那天绍载,我揣著相機與錄音,去河邊找鬼滔蝉。 笑死击儡,一個胖子當著我的面吹牛,可吹牛的內容都是我干的蝠引。 我是一名探鬼主播阳谍,決...
    沈念sama閱讀 40,352評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼螃概!你這毒婦竟也來了边坤?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,257評論 0 276
  • 序言:老撾萬榮一對情侶失蹤谅年,失蹤者是張志新(化名)和其女友劉穎茧痒,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體融蹂,經...
    沈念sama閱讀 45,717評論 1 315
  • 正文 獨居荒郊野嶺守林人離奇死亡旺订,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,894評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了超燃。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片区拳。...
    茶點故事閱讀 40,021評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖意乓,靈堂內的尸體忽然破棺而出樱调,到底是詐尸還是另有隱情,我是刑警寧澤届良,帶...
    沈念sama閱讀 35,735評論 5 346
  • 正文 年R本政府宣布笆凌,位于F島的核電站,受9級特大地震影響士葫,放射性物質發(fā)生泄漏乞而。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,354評論 3 330
  • 文/蒙蒙 一慢显、第九天 我趴在偏房一處隱蔽的房頂上張望爪模。 院中可真熱鬧,春花似錦荚藻、人聲如沸屋灌。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,936評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽共郭。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間落塑,已是汗流浹背纽疟。 一陣腳步聲響...
    開封第一講書人閱讀 33,054評論 1 270
  • 我被黑心中介騙來泰國打工罐韩, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留憾赁,地道東北人。 一個月前我還...
    沈念sama閱讀 48,224評論 3 371
  • 正文 我出身青樓散吵,卻偏偏與公主長得像龙考,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子矾睦,可洞房花燭夜當晚...
    茶點故事閱讀 44,974評論 2 355

推薦閱讀更多精彩內容