首页 > 代码库 > poj 2728 Desert King(最小比率生成树,迭代法)

poj 2728 Desert King(最小比率生成树,迭代法)

引用别人的解释:

题意:有n个村庄,村庄在不同坐标和海拔,现在要对所有村庄供水,只要两个村庄之间有一条路即可,

         建造水管距离为坐标之间的欧几里德距离(好象是叫欧几里德距离吧),费用为海拔之差

         现在要求方案使得费用与距离的比值最小

很显然,这个题目是要求一棵最优比率生成树,



概念

有带权图G, 对于图中每条边e[i], 都有benifit[i](收入)和cost[i](花费), 我们要求的是一棵生成树T, 它使得 ∑(benifit[i]) / ∑(cost[i]), i∈T 最大(或最小).

这显然是一个具有现实意义的问题.

 

解法之一 0-1分数规划

设x[i]等于1或0, 表示边e[i]是否属于生成树.

则我们所求的比率 r = ∑(benifit[i] * x[i]) / ∑(cost[i] * x[i]), 0≤i<m .

为了使 r 最大, 设计一个子问题---> 让 z = ∑(benifit[i] * x[i]) - l * ∑(cost[i] * x[i]) = ∑(d[i] * x[i]) 最大 (d[i] = benifit[i] - l * cost[i]) , 并记为z(l). 我们可以兴高采烈地把z(l)看做以d为边权的最大生成树的总权值.

 


然后明确两个性质:

 1.  z单调递减

  证明: 因为cost为正数, 所以z随l的减小而增大.

 2.  z( max(r) ) = 0

  证明: 若z( max(r) ) < 0, ∑(benifit[i] * x[i]) - max(r) * ∑(cost[i] * x[i]) < 0, 可化为 max(r) < max(r). 矛盾;

          若z( max(r) ) >= 0, 根据性质1, 当z = 0 时r最大.

到了这个地步, 七窍全已打通, 喜欢二分的上二分, 喜欢Dinkelbach的就Dinkelbach.

 

复杂度

时间 O( O(MST) * log max(r) )

空间 O( O(MST) )


迭代+prim

 

#include<stdio.h>#include<math.h>#include<string.h>#include<limits.h>#define MAX 1100double x[MAX],y[MAX],z[MAX];double cost[MAX][MAX],dist[MAX][MAX];int n;double prim(double);int main(void){    int i,j;    while(scanf("%d",&n)&&n){        for(i=1;i<=n;i++){//读取数据            scanf("%lf%lf%lf",&x[i],&y[i],&z[i]);        }        //处理 任意点之间的长度和价值        for(i=1;i<=n;i++){             for(j=i+1;j<=n;j++){                double d=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);                dist[i][j]=dist[j][i]=sqrt(d);                double c=z[i]-z[j];                if(c<0) c=-c;                cost[i][j]=cost[j][i]=c;                            }        }        //判断给定的rate是否在误差之内        double a=0,b=0; //初始r为0        while(1)        {            b = prim(a);            if(fabs(a-b)<1e-4)  break;            a=b;        }                 printf("%.3lf\n",b);              }    return 0;}double prim(double p){    int visit[MAX],father[MAX];    double dis[MAX];    int i,j;    //访问数组初始化    memset(visit,0,sizeof(visit));    visit[1]=1;    //对已加入的最小生成树的顶点集合 计算它的父节点    for(i=2;i<=n;i++){        dis[i]=cost[1][i]-p*dist[1][i];        father[i]=1;    }                    int k=0;    double totalCost=0,totalDist=0;    //prim求最小生成树    for(i=1;i<n;i++){        k=0;        double mincost=INT_MAX;//最大值,最好是这个        for(j=2;j<=n;j++){            if(!visit[j]&&dis[j]<mincost) {                mincost=dis[j];k=j;            }        }                if(k==0) break;        visit[k]=1;        totalCost+=cost[ father[k] ][k];        totalDist+=dist[ father[k] ][k];                for(j=1;j<=n;j++){            double h=cost[k][j]-p*dist[k][j];            if(!visit[j]&&dis[j]>h){                dis[j]=h;                father[j]=k;            }        }            }    return totalCost/totalDist;}