当前位置: 代码迷 >> 综合 >> poj-2728-Desert King-01分数规划+最小生成树
  详细解决方案

poj-2728-Desert King-01分数规划+最小生成树

热度:51   发布时间:2023-12-19 10:45:04.0

01分数规划的题目;

由于是完全图,所以求最小生成树的时候要使用prime算法。

否则的话很容易就超时了。

#include <iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#include<stack>
#include<math.h>
using namespace std;
#define maxm 1100*1100
#define maxn 1100
#define eps 0.000001
#define zero(x) ((fabs(x)<eps?0:x))
#define INF 99999999
struct point
{int x;int y;int h;
}p[maxn];
double dist(int x,int y)
{double xx=0;xx=(p[x].x-p[y].x)*(p[x].x-p[y].x);xx+=(p[x].y-p[y].y)*(p[x].y-p[y].y);xx=xx*1.0;return sqrt(xx);
}
struct list
{double h;double l;double s;friend bool operator < (const list &a,const list &b){return a.s<b.s;}
}maps[maxn][maxn];
int m,n;
double prime(double mid)
{double dis[maxn];int vis[maxn];int i,j;for(i=0;i<=n;i++){dis[i]=INF;vis[i]=0;}for(i=1;i<=n;i++){for(j=1;j<=n;j++){maps[i][j].s=maps[i][j].h-maps[i][j].l*mid;}}dis[1]=0;double ans;ans=0;while(1){double minn=INF;int ip;for(i=1;i<=n;i++){if(vis[i])continue;if(dis[i]<minn){minn=dis[i];ip=i;}}if(zero(minn-INF)==0)break;vis[ip]=1;ans+=minn;for(i=1;i<=n;i++){if(vis[i])continue;dis[i]=min(dis[i],maps[ip][i].s);}}return ans;
}
int main()
{int i,j;while(~scanf("%d",&n)&&n){m=0;for(i=1;i<=n;i++)scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].h);for(i=1;i<=n;i++){for(j=1;j<=n;j++){maps[i][j].h=fabs(p[i].h-p[j].h);maps[i][j].l=dist(i,j);}}double l,r,mid;l=0;r=100;mid=(l+r)/2;while(zero(r-l)>0){mid=(l+r)/2;if(prime(mid)>0)l=mid;else r=mid;}printf("%.3f\n",mid);}return 0;
}