1.Prim算法
priority_queue维护: 可以检测图是不是连通的
struct edge {
int to, cost; // 终点 距离
};
typedef pair<int , int> p; // first 是距离 second 目标顶点
int V; // 顶点数
vector <edge> G[MAX_V]; // 下标起点
bool used[MAX_V]; // 标记是否访问
int Prim() {
priority_queue <p, vector<p>, greater<p> > que;
for (int i = 0; i < V; i++) {
used[i] = false;
}
int cnt = V; // 标记所有点是否被访问
que.push(p(0, 0)); // 从零顶点开始加入集合
int res = 0; // 长度
while (!que.empty() && cnt) {
p temp = que.top();
que.pop();
int v = temp.second;
if (used[v]) continue;
res += temp.first;
// cout << v + 1 << " " << temp.first << endl;
cnt--;
used[v] = true;
for (int u = 0; u < G[v].size(); u++) {
if (!used[G[v][u].to]) {
que.push(p(G[v][u].cost, G[v][u].to));
}
}
}
if (cnt) return -1;
else return res;
}
查找最小值: 默认图时连通的
int cost[MAX_V][MAX_V];
int mincost[MAX_V]; // 从集合x出发到每个顶点的最小权值
bool used[MAX_V];
int V;
int Prim() {
for (int i = 0; i < V; i++) {
used[i] = false;
mincost[i] = INF;
}
mincost[0] = 0; // 从第0个顶点开始 加入集合
int res = 0;
while (true) {
int v = -1;
for (int u = 0; u <V; u++) {
if (!used[u] && (v == -1 || mincost[u] < mincost[v])) {
v = u;
}
}
if (v == -1) break;
used[v] = true;
res += mincost[v];
for (int u = 0; u < V; u++) {
mincost[u] = min(mincost[u], cost[v][u]);
}
}
return res;
}
2.kruskal算法
/*
使用并查集 find(x) init(n) union(x, y)
*/
struct edge {int u, v, cost};
bool cmp(edge e1, edge e2) {
return e1.cost < e2.cost;
}
edge es[MAX_E];
int V, E;
int kruskal() {
sort(es, es + E, cmp);
init(V);
int res = 0;
for (int i = 0; i < E; i++) {
edge e = es[i];
if (find(e.u) != find(e.v)) {
union(e.u, e.v);
res += e.cost;
}
}