poj 2728 Desert King

http://poj.org/problem?id=2728

分数规划+二分   此题的图过于稠密 需要用prime 找最小生成树

代码：

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cmath>

#include<queue>

#include<string.h>



using namespace std;



const double eps=1e-5;

const int N=1001;

const double M=10000000000.0;

struct node

{

    int x,y;

    int h;

}mem[N];//输入点和高度

double d[N][N];//两点间 距离

int cost[N][N];//高度差

bool had[N];

double dist[N];

int n,m;

double findd(int i,int j)//返回距离

{

    return sqrt(1.0*(mem[i].x-mem[j].x)*(mem[i].x-mem[j].x)+

                1.0*(mem[i].y-mem[j].y)*(mem[i].y-mem[j].y));

}

bool prime(double L)//求最小生成树

{

    double ans=0.0;

    memset(had,false,sizeof(had));

    had[1]=true;

    for(int i=2;i<=n;++i)

    dist[i]=cost[1][i]-L*d[1][i];

    for(int w=1;w<n;++w)

    {

        double MIN=M;

        int k=-1;

        for(int i=2;i<=n;++i)

        {

            if(!had[i]&&dist[i]<MIN)

            {

                MIN=dist[i];

                k=i;

            }

        }

        ans+=MIN;

        had[k]=true;

        for(int i=2;i<=n;++i)

        {

            if(!had[i]&&cost[k][i]-L*d[k][i]<dist[i])

            {

                dist[i]=cost[k][i]-L*d[k][i];

            }

        }

    }

    if(ans>0.0)

    return true;

    return false;



}

int main()

{

    //freopen("data.txt","r",stdin);

    while(scanf("%d",&n)!=EOF)

    {

        if(n==0)

        break;

        for(int i=1;i<=n;++i)

        scanf("%d %d %d",&mem[i].x,&mem[i].y,&mem[i].h);

        for(int i=1;i<=n;++i)

        for(int j=i+1;j<=n;++j)

        {

            cost[i][j]=cost[j][i]=abs(mem[i].h-mem[j].h);//求花费

            d[i][j]=d[j][i]=findd(i,j);//求距离

        }

        double l=0.0,r=100.0;

        while(r-l>=eps)//二分缩紧

        {

            double mid=(l+r)/2.0;

            if(prime(mid))

            l=mid;

            else

            r=mid;

        }

        printf("%.3f\n",l);

    }

    return 0;

}