poj 2728 Desert King (最优比率生成树)

链接:poj 2728

题意：有n个村庄，给出每个村庄的坐标和海拔，

benifit为两点之间的水平距离，cost为两点的高度差，

现要求一棵树使得 cost / benift 最小，即求一个最优比例生成树

分析：01规划的应用

设x[i]等于1或0, 表示边取或者不取

则所求的比率 rate = ∑(cost[i] * x[i]) / ∑(benifit[i] * x[i])

设 z = ∑(cost[i] * x[i]) - rate * ∑(benifit[i] * x[i]) 

     = ∑(d[i] * x[i]) 

其中d[i] = cost[i] - rate *benifit[i]

为了使rate最小，则z要最大

则可将d[i]看作边的权值，二分rate,再用prim求最小生成树

#include<stdio.h>
#include<math.h>
#include<string.h>
#define eps 1e-6
#define INF 1e9
typedef struct{
    double x,y,z;
}point;
int n;
double cost[1010][1010],len[1010][1010],map[1010][1010];
double dis(double x1,double y1,double x2,double y2)
{
    return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
double prim(double rate)
{
    int i,j,pos;
    double low[1010];
    bool visit[1010];
    for(i=1;i<=n;i++)
        for(j=i+1;j<=n;j++)
            map[i][j]=map[j][i]=cost[i][j]-rate*len[i][j];  
    memset(visit,0,sizeof(visit));
    pos=1;
    visit[pos]=true;
    for(i=1;i<=n;i++)
        low[i]=map[pos][i];
    double min,s=0;
    for(i=1;i<n;i++){
        min=INF;
        for(j=1;j<=n;j++)
            if(!visit[j]&&low[j]<min){
                pos=j;
                min=low[j];
            }
        if(min==INF)
            break;
        s+=min;
        visit[pos]=true;
        for(j=1;j<=n;j++){
            if(!visit[j]&&map[pos][j]<low[j])
                low[j]=map[pos][j];
        }
    }
    return s;
}
double bin_search(double l,double r)
{
    double mid;
    while(r-l>eps){
        mid=(l+r)/2;
        if(prim(mid)>=0)
            l=mid;
        else
            r=mid;
    }
    return mid;
}
int main()
{
    int i,j;
    point p[1010];
    while(scanf("%d",&n)!=EOF){
        if(n==0)
            break;
        for(i=1;i<=n;i++)
            scanf("%lf%lf%lf",&p[i].x,&p[i].y,&p[i].z);
        for(i=1;i<=n;i++){
            map[i][i]=0.0;
            for(j=i+1;j<=n;j++){
                cost[i][j]=cost[j][i]=fabs(p[i].z-p[j].z);
                len[i][j]=len[j][i]=dis(p[i].x,p[i].y,p[j].x,p[j].y);
            }
        }
        double ans=bin_search(0.0,100.0);
        printf("%.3lf\n",ans);
    }
    return 0;
}