寻宝-卡玛-Prim算法

题目链接： 寻宝

本题是最小生成树的相关题目，最小生成树的题目可以由 Prim 算法 和 Kruskal 进行解答。这里先学习了 最小生成树之prim 的解题思路。
最小生成树是所有节点的最小连通子图， 即：以最小的成本（边的权值）将图中所有节点链接到一起。图中有n个节点，那么一定可以用 n - 1 条边将所有节点连接到一起。
prim算法 是从节点的角度 采用贪心的策略 每次寻找距离 最小生成树最近的节点 并加入到最小生成树中。
prim算法核心就是三步，我称为prim三部曲，大家一定要熟悉这三步，代码相对会好些很多：
第一步，选距离生成树最近节点
第二步，最近节点加入生成树
第三步，更新非生成树节点到生成树的距离（即更新minDist数组），minDist数组 用来记录 每一个节点距离最小生成树的最近距离。

#include <bits/stdc++.h>
using namespace std;

int main(){
    int V, E;
    cin >> V >> E;
    vector<vector<int>> graph(V + 1, vector<int>(V + 1, 10001));
    int s,t,val;
    for(int i = 0; i < E; i++){
        cin >> s >> t >> val;
        graph[s][t] = val;
        graph[t][s] = val;
    }
    vector<int> minDist(V + 1, 10001);
    vector<bool> isInTree(V + 1, false);
    for(int i = 1; i < V; i++){
        int cur = 0;
        int minVal = INT_MAX;
        
        for(int j = 1; j <= V; j++){
            if(!isInTree[j] && minDist[j] < minVal){
                cur = j;
                minVal = minDist[j];
            }
        }
        isInTree[cur] = true;
        
        for(int j = 1; j <= V; j++){
            if(!isInTree[j] && graph[cur][j] < minDist[j]){
                minDist[j] = graph[cur][j];
            }
        }
    }
    int ans = 0;
    for(int i = 2; i < minDist.size(); i++){
        ans += minDist[i];
    }
    cout << ans << endl;
    return 0;
}