题意:给出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;
else map[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;
}