pku 2728 最优比率生成树

题目描述:

给你n个水库,告诉你每个水库的xy坐标(米)和这个水库的花费。

如果两个水库需要连接,连接的费用是这两个水库的高度差的绝对值。

让你连接一些水库,让所有的水库都能到达1号水库并且只有一条通路(就是一棵树)。

然你使每米的花费最少(就是总花费 比 总长度)

解题报告:实际上就求一棵最小比率生成树,黑书上介绍的挺详细的,就是在代码实现上有点困难

有俩种方法实现,一种是二分法,不过在求上下界方面有点难,并且慢很多,一种是用迭代法,标准的01分数规划,不断的逼近迭代最优解;

关于构造式子以及用二分法实现,这个博客上解答的很详细

http://hi.baidu.com/zzningxp/blog/item/b2d1b4ec1f8bbc2262d09fc9.html

#include<stdio.h>
#include<math.h>
#include<string>
#define   Maxvalue 99999999
int n,x[1000],y[1000],h[1000];
double cost[1000][1000],dist[1000][1000],dis[1000];
double prim(double p);
bool flag[1000];
int main()
{
    int i,j;
    double a,b;
    while(scanf("%d",&n)==1&&n)
    {
          for (i=0;i<n;i++)
                scanf("%d%d%d",&x[i],&y[i],&h[i]);
          for (i=0;i<n;i++)
          {
                for (j=i+1;j<n;j++)
                {
                      b=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
                      dist[i][j]=sqrt(b);
                      dist[j][i]=dist[i][j];
                      cost[i][j]=h[i]-h[j];
                      if (cost[i][j]<0)
                            cost[i][j]=-cost[i][j];
                      cost[j][i]=cost[i][j];
                }
          }
          a=0;
          while (1)
          {
                b=prim(a);
                if (fabs(b-a)<0.0001)
                      break;
                a=b;
          }
          printf("%.3lf\n",b);
    }
    return 0;
}
double prim(double p)
{
    int i,j,k,pre[1000]={0};
    double mincost,v,totcost=0,totdist=0;
    dis[0]=0;
	memset(flag,false,sizeof(flag));
    for (j=1;j<n;j++)
          dis[j]=cost[j][0]-p*dist[j][0];
    for (i=1;i<n;i++)
    {
          mincost=Maxvalue;
          for (j=1;j<n;j++)
          {
                if (flag[j]==false && mincost>dis[j])
                {
                      mincost=dis[j];
                      k=j;
                }
          }
          flag[k]=true;
          totcost+=cost[pre[k]][k];
          totdist+=dist[pre[k]][k];
          for (j=0;j<n;j++)
          {
                if (flag[j]==false)
                {
                      if ((v=cost[k][j]-p*dist[k][j])<dis[j])
                      {
                            dis[j]=v;
                            pre[j]=k;
                      }
                }
          }
    }
    return totcost/totdist;
}

  

posted @ 2011-08-20 10:17  枕边梦  阅读(597)  评论(1编辑  收藏  举报