PKU1751Prime算法

原题 http://poj.org/problem?id=1751

//题目分析，先给N个点的坐标，然后给出那些点已经相连，输出最小生成树的边。因为点比较多，所以首选prime算法。
//在输出边的时候可以用一个数组来进行标记。记录与选择的那个点的相连的是那个点。所以这个数组在进行初始化的
//时候应该要把他全部赋值为一。这样当选择了第二个点的时候就可以直接输出。同理。以后每个点的长度发生改变的
//时候都要这个数组的值变为这个点

#include <stdio.h>
#include <stdlib.h>
#include <malloc.h>
#include <ctype.h>
#include <limits.h>
#include <string.h>
#include <string>
#include <algorithm>
#include <iostream>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <math.h>
#include <map>
#include <vector>
using namespace std;
#define INF 9999999.9
#define N 800	
int X[N],Y[N];
double dis[N][N];
double g[N];
int path[N];
int vis[N];
int n,m;

double len(int x1,int y1,int x2,int y2){
	double c;
	c = sqrt((x1-x2+0.0)*(x1-x2+0.0)+(y1-y2+0.0)*(y1-y2+0.0));
	return c;
}

void Prime(){
	//int mark = 2;
	int i,j;
	for(i=1;i<=n;i++){
		g[i] = dis[1][i];
		path[i] = 1;
	}
	vis[1] = 0;
	int k;
	double tmp;
	for(i=1;i<=n;i++){
		tmp = INF;
		for(j=1;j<=n;j++){
			if(tmp>g[j] && vis[j]==0){
				tmp = g[j];
				k = j;
			}
		}
		vis[k] = 1;
		//if(tmp == INF)
		//	break;
		if(dis[path[k]][k] != 0){
			printf("%d %d\n",path[k],k);
		}
		for(j=1;j<=n;j++){
			if(vis[j]==0 && dis[k][j]<g[j]){
				g[j] = dis[k][j];
				path[j] = k;
			}
		}

	}
}

int main(){
	int i,j;

	while(~scanf("%d",&n)){
		memset(vis,0,sizeof(vis));
		memset(g,0,sizeof(g));
		for(i=1;i<=n;i++){
			scanf("%d%d",&X[i],&Y[i]);
		}
		for(i=1;i<=n;i++){
			dis[i][i] = 0;
			for(j=1;j<=i;j++){
				dis[i][j] = len(X[i],Y[i],X[j],Y[j]);
				dis[j][i] = dis[i][j];
			}
		}
		scanf("%d",&m);
		int a,b;
		for(i=1;i<=m;i++){
			scanf("%d%d",&a,&b);
			dis[a][b] = 0;
			dis[b][a] = 0;
		}
		Prime();
	}

	return 0;
}