还是畅通工程
Time Limit : 4000/2000ms (Java/Other) Memory Limit : 65536/32768K (Java/Other) Total Submission(s) : 66 Accepted Submission(s) : 53 Font: Times New Roman | Verdana | Georgia Font Size: ← → Problem Description 某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。 Input 测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。 当N为0时,输入结束,该用例不被处理。 Output 对每个测试用例,在1行里输出最小的公路总长度。 Sample Input 3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0 Sample Output 3 5 Hint Hint Huge input, scanf is recommended. Source 浙大计算机研究生复试上机考试-2006年
这里用的是时间复杂度为n*n的算法,更高效最小堆算法还在学习中。
AC代码:
#include<iostream> #include<cstdio> #include<string> #include<algorithm> #include<cstring> #include<vector> #include<vector> #include<functional> #include<stack> #include<queue> #include<list> using namespace std; #define INF 9999999 int vis[10000]; int map[105][105]; int v[100]; int n; void init() { memset(vis, 0, sizeof(vis)); for (int i = 0;i <= 104;i++) { for (int j = 0;j <= 104;j++) { map[i][j] = INF; } } } int prim() { int temp, mint;int sum = 0; vis[1] = 1; for (int i = 1;i <= n;i++) { v[i] = map[1][i]; } for (int i = 1;i < n;i++) { //cout << sum << endl; mint = INF;temp = 0; for (int j = 1;j <= n;j++) { if (!vis[j] && mint > v[j]) { temp = j; mint = v[j]; } } vis[temp] = 1; sum += mint; for (int j = 1;j <= n;j++) { if (map[temp][j] < v[j] && !vis[j]) { v[j] = map[temp][j]; } } } return sum; } int main() { int x, y, t; while (cin >> n&&n) { init(); for (int i = 1;i <= n*(n - 1) / 2;i++) { cin >> x >> y >> t; if (t < map[x][y]) { map[x][y] = map[y][x] = t; } } cout << prim() << endl; } return 0; }