POJ 1751

来源:互联网 发布:centos 查看软件版本 编辑:程序博客网 时间:2024/04/30 16:00

题意:给出n个点的坐标,然后有给出m个已经相连的路径,求剩下点的最小生成树

思路:prim ,已经相连的点之间的距离赋值为0,每次在加边的时候检测一下这条边是否已经相连,否则就输出。

#include<cmath>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int qq=800;const int MAX=1e8;int n,m,sum;int map[qq][qq];int cost[qq+300];int vis[qq];int pre[qq];int x[qq],y[qq];void prim(){memset(vis,0,sizeof(vis));for(int i=1;i<=n;++i){cost[i]=map[1][i];pre[i]=1;}vis[1]=1;int minx,k;for(int i=1;i<=n;++i){minx=MAX;for(int j=1;j<=n;++j)if(!vis[j] && cost[j]<minx)minx=cost[k=j];if(map[pre[k]][k])printf("%d %d\n",pre[k],k);vis[k]=1;for(int j=1;j<=n;++j)if(map[k][j]<cost[j]){cost[j]=map[k][j];pre[j]=k;}}}int main(){scanf("%d",&n);int i,j,a,b;for(i=1; i<=n; ++i){scanf("%d%d",&x[i],&y[i]);for(j=1; j<i; ++j){if(i==j)map[i][j]=MAX;elsemap[i][j]=map[j][i]=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);}}scanf("%d",&m);for(i=1; i<=m; ++i){scanf("%d%d",&a,&b);map[a][b]=map[b][a]=0;}prim();return 0;}


0 0
原创粉丝点击