POJ 1751 圖論入門第四題

     題目大意：鐵路工人需要建N條路來連接M個村子，有些路已經建好，問你將這些村子連接起來所建路的最小值。最短路問題！

     題目分析：去年在HDU上做過一模一樣的一題，當時用kruskal過的，在POJ上交的時候果斷TLE、無語……！一般情況下，稠密圖我們使用prim算法，稀疏圖使用kruskal算法。當然依題意而定。

     PS：這題需要記錄路徑，記錄路徑一直是我的短項啊。愁啊！

     代碼：

 

#include<cstdio> #include<cmath> #include<cstring> using namespace std; #define init(a,what) memset(a,what,sizeof(a)) #define read freopen("zx.in","r",stdin) #define write freopen("zx.out","w",stdout) const int INF=0x3f3f3f3f; const int MAXN=1000+10; typedef struct { int a,b; }ARR; ARR arr[MAXN]; int nnum,mnum,path[MAXN]; double dist[MAXN][MAXN]; bool vis[MAXN]; double Dist(ARR a,ARR b) { return sqrt((a.a-b.a)*(a.a-b.a)+(a.b-b.b)*(a.b-b.b)); } void prim() { double minn; double lowcost[MAXN]; init(vis,false); vis[1]=true; for(int i=1;i<=nnum;i++) lowcost[i]=dist[1][i]; int i,j,k; for(i=1;i<nnum;i++) { minn=INF; for(j=1;j<=nnum;j++) { if(lowcost[j]<minn && !vis[j]) { minn=lowcost[j], k=j; } } vis[k]=true; for(int t=1;t<=nnum;t++)//這樣來記錄路徑! if(lowcost[t]>dist[k][t] && !vis[t]) { lowcost[t]=dist[k][t]; path[t]=k;//主要就是記錄好father結點 } } } int main() { //read, write; while(scanf("%d",&nnum)!=EOF) { for(int i=1;i<=nnum;i++) path[i]=1;//記錄首節點的path for(int i=1;i<=nnum;i++) scanf("%d%d",&arr[i].a,&arr[i].b); for(int i=1;i<=nnum;i++) for(int j=i;j<=nnum;j++) { if(i==j) dist[i][j]=dist[j][i]=0; else { double cur=Dist(arr[i],arr[j]); dist[i][j]=dist[j][i]=cur; } } scanf("%d",&mnum); for(int i=1;i<=mnum;i++) { int a,b; scanf("%d%d",&a,&b); dist[a][b]=dist[b][a]=0; } prim(); for(int i=1;i<=nnum;i++) if(dist[i][path[i]]) printf("%d %d/n",i,path[i]); } return 0; }