還是暢通工程 HDU
阿新 • • 發佈:2018-12-10
某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。
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 Huge input, scanf is recommended.
最小生成樹水題
#include <stdio.h> #include <algorithm> #include <iostream> #include <cstring> using namespace std; const int maxn = 1010; const int inf = 0x3f3f3f3f; int g[maxn][maxn],d[maxn],vis[maxn],n,m; int prim() { memset(d,inf,sizeof(d)); memset(vis,0,sizeof(vis)); d[1] = 0; int sum = 0; while(1) { int v = -1; for(int u = 1;u <= n;u++) { if(!vis[u] && (v == -1 || d[v] > d[u])) v = u; } if(v == -1) break; vis[v] = 1; sum += d[v]; for(int u = 1;u <= n;u++) { if(!vis[u]) d[u] = min(d[u],g[v][u]); } } return sum; } int main() { while(scanf("%d",&n) && n) { memset(g,inf,sizeof(g)); int m = (n - 1) * n / 2; for(int i = 1;i <= m;i++) { int u,v,w; scanf("%d %d %d",&u,&v,&w); g[u][v] = g[v][u] = w; } int ans = prim(); printf("%d\n",ans); } return 0; }