hdu1875: 畅通工程再续

hud1875: http://acm.hdu.edu.cn/showproblem.php?pid=1875
题意：给出c个点的坐标，若点与点之间的距离大于1000或小于10则不能建桥，建桥费用为每米100元，求使得所有点能相通的最小费用，若不能相通，则输出“oh!"。
解法：最小生成树prim。
code：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int inf=1<<27;
double v[110][110];
int sign[110],x[110],y[110];
double solve(int x1,int y1,int x2,int y2)  //求点与点的距离
{
    double m;
    m=sqrt(1.0*((x2-x1)*(x2-x1)+(y2-y1)*(y2-y1)));
    return m;
}
int main()
{
    int t,c,u;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&c);
        for(int i=0;i<c;i++)
            scanf("%d%d",&x[i],&y[i]);
        for(int i=0;i<c;i++)
        for(int j=0;j<c;j++)
        {
            v[i][j]=inf;v[j][i]=inf;
        }
        for(int i=0;i<c-1;i++)
        {
            for(int j=i+1;j<c;j++)
            {
                double a=solve(x[i],y[i],x[j],y[j]);
                if(a<10||a>1000)
                {
                    v[i][j]=inf;v[j][i]=inf;
                }
                else
                {
                    v[i][j]=100.0*a;
                    v[j][i]=100.0*a;
                }
            }
        }
        double ans=0,min;
        int l=0;
        memset(sign,0,sizeof(sign));
        for(int i=0;i<c;i++)   //prim
        {
            sign[0]=1;min=inf;
            for(int j=0;j<c;j++)
            {
                if(!sign[j]&&v[0][j]<min)
                {
                    min=v[0][j];u=j;
                }
            }
            sign[u]=1;
            if(min==inf)       //若min=inf,分两种情况：最小生成树已建成；有一个点与其他点都不能连通
            {
                for(int j=0;j<c;j++)
                    if(sign[j]==0)   //说明有点未加入，最小生成树未建成
                        l=1;
                if(l==1)
                    break;
            }
            else
                ans=ans+min;
            for(int j=0;j<c;j++)
            {
                if(v[u][j]<v[0][j])
                    v[0][j]=v[u][j];
            }
        }
        if(l==1)
            printf("oh!\n");
        else
            printf("%.1lf\n",ans);
    }
}
/*input:
2
2
10 10
20 20
3
1 1
2 2
1000 1000
output:
1414.2
oh!*/

转载于:https://www.cnblogs.com/acmjun/archive/2012/07/25/2608652.html