[HDU](1233)还是通畅工程 ---最小生成树(图)
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
当N为0时,输入结束,该用例不被处理。
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
3
5
Huge input, scanf is recommended.
AC代码:Prim算法,模板题
#include<iostream> #include<cstring> using namespace std; int mmap[1005][1005]; bool vis[1005]; int dist[1005]; int n; #define INF 0x3f3f3f3f int prime(int x) { for(int i=1;i<=n;i++) dist[i]=mmap[x][i]; int mmin; int k; int sum=0; for(int i=1;i<=n;i++) { mmin=INF; for(int j=1;j<=n;j++) { if(!vis[j] && dist[j]<mmin) { mmin=dist[j]; k=j; } } vis[k]=true; sum+=mmin; for(int j=1;j<=n;j++) { if(!vis[j] && mmap[k][j]<dist[j]) dist[j]=mmap[k][j]; } } return sum; } int main() { int u,v,w; while(cin>>n&&n) { memset(mmap,0,sizeof(mmap)); memset(vis,false,sizeof(vis)); memset(dist,0,sizeof(dist)); for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { if(i==j) mmap[i][j]=0; else mmap[i][j]=INF; } } int m=n*(n-1)/2; cout<<m<<endl; while(m--) { cin>>u>>v>>w; if(w<mmap[u][v]) { mmap[u][v]=mmap[v][u]=w; } } vis[1]=true; cout<<prime(1)<<endl; } return 0; }