最小生成树 Prim算法(用于稀疏图): 时间复杂度O(n^2+m) 朴素版prim算法,一般应用于稀疏图,形式上和dijkstra算法很相似。 初始化d数组,将1节点加入到最小生成树节点集合中。 迭代n-1次,每次在S中寻找到T集合的最短距离,用这个节点更新其他非最小生成树节点。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 int Prim () { int res = 0 ; memset (dis,0x3f ,sizeof dis); dis[1 ] = 0 ; for (int i = 0 ;i < n; i++){ int t = -1 ; for (int j = 1 ; j <= n; j++){ if (!vis[j] && (t==-1 || dis[j] < dis[t])) t = j; } if (dis[t] == INF) return INF; res += dis[t]; vis[t] = true ; for (int i = 1 ; i <= n; i++){ if (!vis[i]) dis[i] = min (dis[i],g[t][i]); } } return res; }
Prim算法求最小生成树 给定一个 n个点 m 条边的无向图,图中可能存在重边和自环,边权可能为负数。
求最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible
。
给定一张边带权的无向图 G=(V,E),其中 V 表示图中点的集合,E 表示图中边的集合,n=|V|,m=|E|。
由 V 中的全部 n 个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G 的一棵生成树,其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。
输入格式 第一行包含两个整数 n 和 m。
接下来 m 行,每行包含三个整数 u,v,w表示点 u 和点 v 之间存在一条权值为 w 的边。
输出格式 共一行,若存在最小生成树,则输出一个整数,表示最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible
。
数据范围 1≤n≤500 1≤m≤1e5, 图中涉及边的边权的绝对值均不超过 10000。
输入样例: 1 2 3 4 5 6 4 5 1 2 1 1 3 2 1 4 3 2 3 2 3 4 4
输出样例:
代码 个人代码 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 #include <bits/stdc++.h> using namespace std;const int N = 510 ;int g[N][N];int dis[N],st[N];int n;void Prime () { int res = 0 ; dis[1 ] = 0 ; for (int i=0 ;i<n;i++){ int t = -1 ; for (int j=1 ;j<=n;j++){ if (!st[j] && (t == -1 || dis[t] > dis[j])) { t = j; } } st[t] = 1 ; if (dis[t] == 0x3f3f3f3f ) { cout<<"impossible" <<endl; return ; } res += dis[t]; for (int j=1 ;j<=n;j++){ dis[j] = min (dis[j], g[t][j]); } } cout<<res<<endl; return ; } int main () { memset (dis, 0x3f ,sizeof dis); memset (g,0x3f ,sizeof g); int m;cin>>n>>m; while (m--){ int a,b,c; cin>>a>>b>>c; g[a][b] = min (g[a][b], c); g[b][a] = min (g[b][a], c); } Prime (); }
板子 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 #include <cstring> #include <iostream> #include <algorithm> using namespace std;const int N = 510 , INF = 0x3f3f3f3f ;int n, m;int g[N][N];int dist[N];bool st[N];int prim () { memset (dist, 0x3f , sizeof dist); int res = 0 ; for (int i = 0 ; i < n; i ++ ) { int t = -1 ; for (int j = 1 ; j <= n; j ++ ) if (!st[j] && (t == -1 || dist[t] > dist[j])) t = j; if (i && dist[t] == INF) return INF; if (i) res += dist[t]; st[t] = true ; for (int j = 1 ; j <= n; j ++ ) dist[j] = min (dist[j], g[t][j]); } return res; } int main () { scanf ("%d%d" , &n, &m); memset (g, 0x3f , sizeof g); while (m -- ) { int a, b, c; scanf ("%d%d%d" , &a, &b, &c); g[a][b] = g[b][a] = min (g[a][b], c); } int t = prim (); if (t == INF) puts ("impossible" ); else printf ("%d\n" , t); return 0 ; }
Kruskal算法(用于稠密图): 时间复杂度O(mlogm) 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 int n, m; int p[N]; struct Edge { int a, b, w; bool operator < (const Edge &W)const { return w < W.w; } }edges[M]; int find (int x) { if (p[x] != x) p[x] = find (p[x]); return p[x]; } int kruskal () { sort (edges, edges + m); for (int i = 1 ; i <= n; i ++ ) p[i] = i; int res = 0 , cnt = 0 ; for (int i = 0 ; i < m; i ++ ) { int a = edges[i].a, b = edges[i].b, w = edges[i].w; a = find (a), b = find (b); if (a != b) { p[a] = b; res += w; cnt ++ ; } } if (cnt < n - 1 ) return INF; return res; }
Kruskal算法求最小生成树 给定一个 n 个点 m 条边的无向图,图中可能存在重边和自环,边权可能为负数。
求最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible
。
给定一张边带权的无向图 G=(V,E)其中 V 表示图中点的集合,E 表示图中边的集合,n=|V|,m=|E|。
由 V 中的全部 n 个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G 的一棵生成树,其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。
输入格式 第一行包含两个整数 n 和 m。
接下来 m 行,每行包含三个整数 u,v,w表示点 u 和点 v 之间存在一条权值为 w 的边。
输出格式 共一行,若存在最小生成树,则输出一个整数,表示最小生成树的树边权重之和,如果最小生成树不存在则输出 impossible
。
数据范围 1≤n≤1e5, 1≤m≤2∗1e5, 图中涉及边的边权的绝对值均不超过 1000。
输入样例: 1 2 3 4 5 6 4 5 1 2 1 1 3 2 1 4 3 2 3 2 3 4 4
输出样例:
代码 个人代码 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 #include <bits/stdc++.h> using namespace std;const int N = 10 ;struct line { int a, b, w; line () {} line (int a, int b, int w): a (a), b (b), w (w) {} bool operator <(const line k) { return w < k.w; } }; vector<line> allLine; int fa[N];int find (int x) { if (fa[x] != x) fa[x] = find (fa[x]); return fa[x]; } int main () { int n, m; cin >> n >> m; while (m--) { int a, b, v; cin >> a >> b >> v; allLine.push_back (line (a, b, v)); } sort (allLine.begin (),allLine.end ()); for (int i=1 ;i<=n;i++){ fa[i] = i; } int cnt = 0 ; int res = 0 ; for (int i=0 ;i < allLine.size ();i++){ line a = allLine[i]; if (find (a.a) != find (a.b)) { fa[find (a.a)] = find (a.b); cnt++; res+=a.w; if (cnt == n - 1 ) { cout<<res<<endl; return 0 ; } } } cout<<"impossible" <<endl; return 0 ; }
板子 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 #include <cstring> #include <iostream> #include <algorithm> using namespace std;const int N = 100010 , M = 200010 , INF = 0x3f3f3f3f ;int n, m;int p[N];struct Edge { int a, b, w; bool operator < (const Edge &W)const { return w < W.w; } }edges[M]; int find (int x) { if (p[x] != x) p[x] = find (p[x]); return p[x]; } int kruskal () { sort (edges, edges + m); for (int i = 1 ; i <= n; i ++ ) p[i] = i; int res = 0 , cnt = 0 ; for (int i = 0 ; i < m; i ++ ) { int a = edges[i].a, b = edges[i].b, w = edges[i].w; a = find (a), b = find (b); if (a != b) { p[a] = b; res += w; cnt ++ ; } } if (cnt < n - 1 ) return INF; return res; } int main () { scanf ("%d%d" , &n, &m); for (int i = 0 ; i < m; i ++ ) { int a, b, w; scanf ("%d%d%d" , &a, &b, &w); edges[i] = {a, b, w}; } int t = kruskal (); if (t == INF) puts ("impossible" ); else printf ("%d\n" , t); return 0 ; }