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
#include
#include
#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;ieps){
        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;
}