poj2728 Desert King（最优比率生成树）

题目链接：点击打开链接

题意描述：在一个国家的城镇之间建水渠，把所有的城镇链接起来（n-1条边），两个城镇之间的长度为两点之间的直线距离，花费为两个城镇位置的高度之差。问花费之和/长度之和最小为多少？

解题思路：最优比率生成树点击打开链接

代码：

#include <cstdio>
#include <cmath>
#include <cstring>
#define MAXN 1010
#define eps 1e-5
#define INF 0x7fffffff
using namespace std;
struct point{
    double x,y;
    double h;
}p[MAXN];
double cost[MAXN][MAXN],dist[MAXN][MAXN],g[MAXN][MAXN],dis[MAXN];
bool vis[MAXN];
int pre[MAXN];
int n;
double prime(double x){
    for(int i=1;i<=n;++i)
        for(int j=i+1;j<=n;++j){
            g[j][i]=g[i][j]=cost[i][j]-dist[i][j]*x;
        }
    for(int i=1;i<=n;++i) g[i][i]=0;
    double tolcost=0,toldist=0;
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;++i) dis[i]=INF;
    vis[1]=true;dis[1]=0.0;
    int ct=1;
    double mindis;
    int k;
    for(int t=1;t<=n-1;t++){
        mindis=INF;
        for(int i=1;i<=n;++i){
            if(!vis[i]){
                if(dis[i]>g[ct][i]){ dis[i]=g[ct][i]; pre[i]=ct;}
                if(dis[i]<mindis){ mindis=dis[i];k=i; }
            }
        }
        ct=k; vis[ct]=true;
        tolcost+=cost[pre[ct]][ct];
        toldist+=dist[pre[ct]][ct];
    }
    return tolcost/toldist;
}
double getdist(point a,point b){
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
int main(){
    while(scanf("%d",&n),n){
        for(int i=1;i<=n;++i) scanf("%lf%lf%lf",&p[i].x,&p[i].y,&p[i].h);
        for(int i=1;i<=n;++i) for(int j=i+1;j<=n;++j){
            cost[i][j]=cost[j][i]=abs((double)p[i].h-p[j].h);
            dist[i][j]=dist[j][i]=getdist(p[i],p[j]);
        }
        double ans=0,tans;
        while(1){
            tans=prime(ans);
            if(tans-ans<eps&&ans-tans<eps) break;
            ans=tans;
        }
        printf("%.3lf\n",ans);
    }
    return 0;
}