N - 畅通工程再续

prime算法，注意初始化

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=1000+10;
const double INF=1000000.0;
double dis[maxn];
double G[maxn][maxn];
int vis[maxn];
struct point{
	double x,y;
}p[maxn];
double prime(int n)
{
	memset(vis,0,sizeof(vis));
	vis[1]=1; 
	double ans=0;
	for(int i=1;i<=n;i++)
	{
		dis[i]=G[1][i];
	}
	for(int i=1;i<n;i++)
	{
		int k=-1;double minn=INF;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&dis[j]<minn)
			minn=dis[k=j];
		}
		vis[k]=1;ans+=minn;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&k!=j&&dis[j]>G[k][j])
				dis[j]=G[k][j];
		}
	}
	return ans;
}
bool ok(int n)
{
	for(int i=1;i<=n;i++)
		if(!vis[i])	return false;
	return true;
}
double len(point a,point b)
{
	double x=a.x-b.x;
	double y=a.y-b.y;
	return sqrt(x*x+y*y);
}
void init(int n)
{
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			G[i][j]=i==j?0:INF;
}
int main()
{
	int t,n;
	//freopen("N.txt","r",stdin);
	scanf("%d",&t);
	while(t--)
	{		
		scanf("%d",&n);
		init(n);
		for(int i=1;i<=n;i++)
		{
			scanf("%lf %lf",&p[i].x,&p[i].y);
			for(int j=1;j<i;j++)
			{
				double Len=len(p[i],p[j]);
				if(Len>=10&&Len<=1000)	G[i][j]=G[j][i]=Len;	
			}				
		}
		double ans=prime(n);
		if(ans!=INF)	printf("%.1lf\n",ans*100);
		else	printf("oh!\n");
	}
	return 0;
}