HDOJ1785 畅通工程再续（最小生成树）

Prim算法

/*HDOJ2544
作者：陈佳润
2013-04-09
*/
#include<stdio.h>
#include<math.h>
int N;//表示点的个数
double map[105][105];//表示距离矩阵
bool judge[105];//表示判断表，判断是否在集合内

double Prim(){ //Prim 算法
    int i,j,tnum,k; 
    double sum=0,tmin; 
    //初始化判断数组 
    for(i=1;i<=N;i++)  
        judge[i]=false; 
    judge[1]=true; 
    //Prim 
    for(k=1;k<N;k++){ 
        tmin=0x3f3f3f3f; 
        tnum=0; 
        for(i=1;i<=N;i++){ 
            for(j=2;j<=N;j++){ 
                if(!judge[j]&&judge[i]) 
                    if(map[i][j]<tmin){ 
                        tmin=map[i][j]; 
                        tnum=j; 
                    }//end if  
            }//end j 
        }// end i 
        judge[tnum]=true; 
        sum+=tmin; 
    }//end k 
    return sum; 
}  

int main(){
	int x[105],y[105],t,i,j;
	double result;
	//freopen("1.txt","r",stdin);
	scanf("%d",&t);
	while(t--){
		scanf("%d",&N);
		for(i=1;i<=N;i++)
			scanf("%d%d",&x[i],&y[i]);
		for(i=1;i<=N;i++){//将点的信息转化成距离矩阵
			for(j=1;j<=N;j++){
				map[i][j]=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
				if(map[i][j]<10||map[i][j]>1000)
					map[i][j]=0x3f3f3f3f;
			}
		}

		result=Prim();
		result*=100;
		if(result<0x3f3f3f3f)//如果有路
			printf("%.1lf\n",result);
		else
			printf("oh!\n");
		
	}
	return 0;
}