HDU 1233 還是暢通工程(最小生成樹prim 模板)
阿新 • • 發佈:2018-11-12
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
最小生成樹的模板題,可以用prim和kruaskal,這裡我用是prim
#include<iostream> #include<cstdio> #include<cstring> #define inf 0x3f3f3f3f using namespace std; //最小生成樹 int map[105][105],vis[105],dis[105]; int n; void init() { for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) map[i][j]=(i==j)?0:inf; } void prim() { memset(dis,inf,sizeof(dis)); memset(vis,0,sizeof(vis)); int sum=0; int cnt=1; int i,j; vis[1]=1; for(i=1;i<=n;i++) dis[i]=map[1][i]; for(j=1;j<=n;j++) { int minn=inf; int k=1; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]<minn) { minn=dis[i]; k=i; } } sum+=dis[k]; vis[k]=1; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]>map[k][i]) { dis[i]=map[k][i]; //vis[i]=1; } } } cout<<sum<<endl; //輸出最小生成樹的權值和 } int main() { while(scanf("%d",&n)&&n) { init(); //初始化map矩陣 int a,b,w,i,j; for(i=1;i<=n*(n-1)/2;i++) //輸入矩陣 { scanf("%d%d%d",&a,&b,&w); if(w<map[a][b]) map[a][b]=map[b][a]=w; } prim(); } return 0; }#include<iostream> #include<cstdio> #include<cstring> #define inf 0x3f3f3f3f using namespace std; //最小生成樹 int map[105][105],vis[105],dis[105]; int n; void init() { for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) map[i][j]=(i==j)?0:inf; } void prim() { memset(dis,inf,sizeof(dis)); memset(vis,0,sizeof(vis)); int sum=0; int cnt=1; int i,j; vis[1]=1; for(i=1;i<=n;i++) dis[i]=map[1][i]; for(j=1;j<=n;j++) { int minn=inf; int k=1; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]<minn) { minn=dis[i]; k=i; } } sum+=dis[k]; vis[k]=1; for(i=1;i<=n;i++) { if(!vis[i]&&dis[i]>map[k][i]) { dis[i]=map[k][i]; //vis[i]=1; } } } cout<<sum<<endl; //輸出最小生成樹的權值和 } int main() { while(scanf("%d",&n)&&n) { init(); //初始化map矩陣 int a,b,w,i,j; for(i=1;i<=n*(n-1)/2;i++) //輸入矩陣 { scanf("%d%d%d",&a,&b,&w); if(w<map[a][b]) map[a][b]=map[b][a]=w; } prim(); } return 0; }