当前位置: 代码迷 >> 综合 >> poj 2728 Desert King 参数搜索解最优比例生成树
  详细解决方案

poj 2728 Desert King 参数搜索解最优比例生成树

热度:50   发布时间:2024-01-19 05:45:10.0

题意:

图中每条边有两个权值(cost,len),求其一棵生成树,使sum(cost)/sum(len)最小。

分析:

转化为求边权为s0*len-cost的最大生成树+牛顿迭代。s0为具有单调性迭代系数。

代码:

//poj 2728
//sep9
#include <iostream>
#include <cmath>
#include <vector>
using namespace std;
const int maxN=1024;
struct P
{int x,y,z;
}v[maxN];int cost[maxN][maxN];
double len[maxN][maxN];
double dis[maxN];
vector<int> g[maxN];
int vis[maxN];
int prev[maxN];
double s0,s1;
int n;double get_v()
{int sum_cost=0;double sum_len=0;for(int i=0;i<n;++i)if(g[i].size()){int j=g[i][0];sum_cost+=cost[i][j];sum_len+=len[i][j];}return sum_cost/sum_len;
}void prim()
{for(int i=0;i<n;++i)g[i].clear();for(int i=0;i<n;++i)dis[i]=-1e12;	dis[0]=0;memset(vis,0,sizeof(vis));for(int t=0;t<n;++t){double maxx=-1e12;int maxi=-1;for(int i=0;i<n;++i){if(vis[i]==0&&dis[i]>maxx){maxi=i;maxx=dis[i];}}	vis[maxi]=1;for(int i=0;i<n;++i)if(vis[i]==0)if(s0*len[maxi][i]-cost[maxi][i]>=dis[i]){dis[i]=s0*len[maxi][i]-cost[maxi][i];prev[i]=maxi;}		}for(int i=1;i<n;++i)g[i].push_back(prev[i]);
}int main()
{while(scanf("%d",&n)==1&&n){for(int i=0;i<n;++i)scanf("%d%d%d",&v[i].x,&v[i].y,&v[i].z);	for(int i=0;i<n;++i)for(int j=i+1;j<n;++j){double dx=(v[i].x-v[j].x)*1.0;double dy=(v[i].y-v[j].y)*1.0;len[i][j]=len[j][i]=sqrt(dx*dx+dy*dy);cost[i][j]=cost[j][i]=abs(v[i].z-v[j].z);}for(int i=0;i<n;++i) g[i].clear();for(int i=0;i<n-1;++i)g[i].push_back(i+1);s0=get_v();while(1){prim();s1=get_v();if(fabs(s0-s1)<1e-5)break;	s0=s1;}	printf("%.3lf\n",s0+1e-9);}return 0;	
}