hdu 1875 畅通工程再续

是一道最小生成树的题目,套用prim算法的模板就可以了

#include<stdio.h>
#include<string.h>
#include<math.h>
#define arr 105
#define inf 200000000.0
double g[arr][arr],dis[arr];
struct node
{
    int x,y;
}a[arr];
double Dis(int i,int j)
{
    double tmp=hypot(a[i].x-a[j].x,a[i].y-a[j].y);
    if(tmp<10.0||tmp>1000.0) 
		tmp=inf; 
    return tmp; 
}        
double Prim(int st,int m){
    int i,Time,mark[arr];
    for(i=1;i<=m;i++) 
	{
		dis[i]=g[st][i];
		mark[i]=0;
	}
    dis[st]=inf;
 	mark[st]=1;
    double ans=0;
    for(Time=1;Time<m;Time++)
	{
        double min= inf;
        int index=-1;
        for(i=1;i<=m;i++)
            if(!mark[i]&&dis[i]<min) 
			{
				min=dis[i];
				index=i;
			}
        if(index==-1) 
			return -1; 
        mark[index]=1;
        ans+= min;
        for(i=1;i<=m;i++)
            if(!mark[i]&&g[index][i]<dis[i])
                dis[i]=g[index][i];
    } 
    return ans*100.0;
}            
int main()
{
    int T,i,j,n,x,y;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)for(j=1;j<=n;j++)g[i][j]=inf;
        for(i=1;i<=n;i++)
            scanf("%d%d",&a[i].x,&a[i].y);
        for(i=1;i<n;i++)
            for(j=i+1;j<=n;j++)
                g[i][j]=g[j][i]= Dis(i,j);      
        double ans=Prim(1,n); //Prim()算法求最小生成树;
        if(ans==-1) 
			printf("oh!");
        else 
			printf("%.1lf\n",ans);
    }
}           


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值