代碼隨想錄打卡:最小生成樹

找到一條能夠把所有點連接起來的最短路徑

prim算法

用一個數(shù)組minDist來記錄每一個節(jié)點距離最小生成樹的最近距離,

用一個boolean數(shù)組visited記錄節(jié)點是否加入最小生成樹

算法流程:

  1. 選擇離最小生成樹最近的(minDist值最斜栌洹),還沒有加入最小生成樹(visited為false)的節(jié)點

  2. 標記節(jié)點加入最小生成樹(visited為true)

  3. 更新minDist中還沒有加入最小生成樹的節(jié)點的值

    用什么更新? --因為剛剛加入新的節(jié)點饺藤,用新節(jié)點到這些節(jié)點的距離更新

    (minDist[不是生成樹的節(jié)點] = Math.min(graph[剛加入的節(jié)點][不是生成樹的節(jié)點], minDist[不是生成樹的節(jié)點]))

import java.util.*;

public class Main{
    public static int V;
    public static int E;
    public static int[][] graph;
    public static boolean[] visited;
    //minDist數(shù)組 用來記錄 每一個節(jié)點距離最小生成樹的最近距離
    public static int[] minDist;

    public static void main (String[] args) {
        /* code */
        Scanner in = new Scanner(System.in);
        V = in.nextInt();
        E = in.nextInt();

        graph = new int[V+1][V+1];
        visited = new boolean[V+1];
        minDist = new int[V+1];
        Arrays.fill(minDist, 10001);

        //因為val>=0,所以要做一個二維數(shù)組的初始化
        for(int i=1; i<=V; i++){
            Arrays.fill(graph[i], -1);
        }

        for(int i=0; i<E; i++){
            int start = in.nextInt();
            int end = in.nextInt();
            int val = in.nextInt();
            graph[start][end] = val;
            //注意是無向圖
            graph[end][start] = val;
        }

        prim();
        
        int result = 0;
        for(int i=2; i<=V; i++){
            result+=minDist[i];
        }
        System.out.println(result);
    }
    public static void prim(){
        // i循環(huán)的是次數(shù)包斑,一共要做V次(把v個節(jié)點加入生成樹)
        for(int i=0; i<V; i++){
            int min_val = Integer.MAX_VALUE;
            int cur = -1;
            //選擇距離最小生成樹最近的點加入最小生成樹
            for(int j=1; j<=V; j++){
                if(!visited[j]&&minDist[j]<min_val){
                    min_val = minDist[j];
                    cur = j;
                }
            }
            //加入
            visited[cur]=true;
            //更新沒加入最小生成樹的節(jié)點到達最小生成樹的距離
            for(int j=1; j<=V; j++){
                if(!visited[j]&&graph[cur][j]>=0)
                    minDist[j] = Math.min(graph[cur][j], minDist[j]);
            }
        }
    }
}

Kruskal

用到并查集的知識

  1. 用LinkedList記錄所有的邊,并按照權(quán)值排序

  2. 遍歷有序(升序)的邊:

    • 如果邊的兩個點不在同一個集合涕俗,則加入最小生成樹(用result記錄樹的長度罗丰,所以result要+邊長,且要把兩個點并為同一個集合)
    • 如果在同一個集合再姑,直接跳過這個邊萌抵,看下一條邊
import java.util.*;

public class Main{
    public static int V;
    public static int E;
    public static LinkedList<int[]> edges;
    public static int[] father;
    public static int result;

    public static void main (String[] args) {
        /* code */
        Scanner in = new Scanner(System.in);
        V = in.nextInt();
        E = in.nextInt();

        edges = new LinkedList<>();
        father = new int[V+1];
        result = 0;
        for(int i=1; i<=V; i++){
            father[i] = i;
        }
        for(int i=0; i<E; i++){
            int start = in.nextInt();
            int end = in.nextInt();
            int val = in.nextInt();
            edges.add(new int[]{start, end, val});
        }
        //List自定義排序規(guī)則!T啤绍填!
        edges.sort(Comparator.comparingInt(o -> o[2]));
        kruskal();
        System.out.println(result);
    }
    public static int find(int u){
        if(u==father[u]) return u;
        else return father[u] = find(father[u]);
    }
    public static boolean isSame(int u, int v){
        u = find(u);
        v = find(v);
        return u==v;
    }

    public static void join(int u, int v) {
        u = find(u);
        v = find(v);
        father[v] = u;
    }

    public static void kruskal(){
        //V個節(jié)點,加V-1條邊就行了栖疑,所以需要重復(fù)V-1次
        for(int i=1; i<V; i++){
            int[] edge = edges.removeFirst();
            while(isSame(edge[0], edge[1])){
                edge = edges.removeFirst();
            }
            result += edge[2];
            join(edge[0], edge[1]);
        }
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末讨永,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子遇革,更是在濱河造成了極大的恐慌卿闹,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,454評論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件萝快,死亡現(xiàn)場離奇詭異锻霎,居然都是意外死亡,警方通過查閱死者的電腦和手機杠巡,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,553評論 3 385
  • 文/潘曉璐 我一進店門量窘,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人氢拥,你說我怎么就攤上這事蚌铜。” “怎么了嫩海?”我有些...
    開封第一講書人閱讀 157,921評論 0 348
  • 文/不壞的土叔 我叫張陵冬殃,是天一觀的道長。 經(jīng)常有香客問我叁怪,道長审葬,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,648評論 1 284
  • 正文 為了忘掉前任奕谭,我火速辦了婚禮涣觉,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘血柳。我一直安慰自己官册,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 65,770評論 6 386
  • 文/花漫 我一把揭開白布难捌。 她就那樣靜靜地躺著膝宁,像睡著了一般鸦难。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上员淫,一...
    開封第一講書人閱讀 49,950評論 1 291
  • 那天合蔽,我揣著相機與錄音,去河邊找鬼介返。 笑死拴事,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的圣蝎。 我是一名探鬼主播挤聘,決...
    沈念sama閱讀 39,090評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼捅彻!你這毒婦竟也來了组去?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,817評論 0 268
  • 序言:老撾萬榮一對情侶失蹤步淹,失蹤者是張志新(化名)和其女友劉穎从隆,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體缭裆,經(jīng)...
    沈念sama閱讀 44,275評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡键闺,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,592評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了澈驼。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片辛燥。...
    茶點故事閱讀 38,724評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖缝其,靈堂內(nèi)的尸體忽然破棺而出挎塌,到底是詐尸還是另有隱情,我是刑警寧澤内边,帶...
    沈念sama閱讀 34,409評論 4 333
  • 正文 年R本政府宣布榴都,位于F島的核電站,受9級特大地震影響漠其,放射性物質(zhì)發(fā)生泄漏嘴高。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 40,052評論 3 316
  • 文/蒙蒙 一和屎、第九天 我趴在偏房一處隱蔽的房頂上張望拴驮。 院中可真熱鬧,春花似錦柴信、人聲如沸套啤。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,815評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽纲岭。三九已至,卻和暖如春线罕,著一層夾襖步出監(jiān)牢的瞬間止潮,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,043評論 1 266
  • 我被黑心中介騙來泰國打工钞楼, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留喇闸,地道東北人。 一個月前我還...
    沈念sama閱讀 46,503評論 2 361
  • 正文 我出身青樓询件,卻偏偏與公主長得像燃乍,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子宛琅,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,627評論 2 350

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