傻傻分不清、

想更靠近、

POJ 1751

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


阅读更多
版权声明:吸猫大法、 https://blog.csdn.net/sasuke__/article/details/52381154
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭