poj1751——Highways

题意：给定N个城市的坐标，然后下面给出M个已经连接好的城市标号，求他们之间的最小生成树，其中，已连接的话，不必输出。

思路：将已给的城市之间距离标为0，输出的时候，城市之间为零的就不必输出。

#include<iostream> #include<cstdio> #include<cstring> #include<math.h> using namespace std; class tt { public: int x,y; }town[760]; int g[760][760]; int n,m; void solve_dis() { int i,j; for(i=1;i<n;i++) { for(j=i+1;j<=n;j++) { g[i][j]=g[j][i]=((town[i].x-town[j].x)*(town[i].x-town[j].x)+(town[i].y-town[j].y)*(town[i].y-town[j].y)); } } } void prim() { int i,j,k,maxcost =999999999,min; int dist[760],pre[760]; bool vis[760]; memset(vis,true,sizeof(vis)); for(i=1;i<=n;i++) { dist[i]=maxcost; pre[i]=1; } vis[1]=false;dist[1]=0; for(i=1;i<=n;i++) { min=maxcost;k=1; for(j=1;j<=n;j++) if(vis[j]&&dist[j]<min) { k=j; min=dist[j]; } vis[k]=false; for(j=1;j<=n;j++) if(dist[j]>g[k][j]&&vis[j]) { dist[j]=g[k][j]; pre[j]=k;} } for(i=2;i<=n;i++) { if(g[pre[i]][i]!=0) { printf("%d %d\n",i,pre[i]); } } } int main() { int i,x,y; scanf("%d",&n); for(i=1;i<=n;i++) scanf("%d%d",&town[i].x,&town[i].y); solve_dis(); scanf("%d",&m); for(i=0;i<m;i++) { scanf("%d%d",&x,&y); g[x][y]=g[y][x]=0; } prim(); }