题目链接: 寻宝
本题是最小生成树的相关题目,最小生成树的题目可以由 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;
}