POJ.2728.Desert King(最优比率生成树 Prim 01分数规划 二分/Dinkelbach迭代)

题目链接

\(Description\)

将n个村庄连成一棵树，村之间的距离为两村的欧几里得距离，村之间的花费为海拔z的差，求花费和与长度和的最小比值

\(Solution\)

二分，假设mid为可行的某一生成树的解，则应有 \((∑cost)/(∑dis) = mid\)

变形得 \(\sum(cost-mid*dis) = 0\)

取cost-mid*dis为边权，Prim求最小生成树(即尽可能满足mid)

若\(\sum(cost-mid*dis) > 0\)，说明怎么也满足不了mid，mid不是可行解 偏小；若 < 0，则存在某些生成树满足条件，还可以更优

若 = 0，那么就是最小值了

1.二分

//19100K 1235MS
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
const int N=1005;
const double eps=1e-4,INF=1e8;
int n,x[N],y[N],z[N],cost[N][N];
double dis[N][N],e[N][N],d[N];
bool vis[N];
inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
inline double Calc(int i,int j) {return sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+1.0*(y[i]-y[j])*(y[i]-y[j]));}
bool Check(double x)
{
	for(int i=1; i<=n; ++i)
		for(int j=i+1; j<=n; ++j)
			e[j][i]=e[i][j]=1.0*cost[i][j]-x*dis[i][j];
	double res=0;//Prim
	memset(vis,0,sizeof vis);
	for(int i=2; i<=n; ++i) d[i]=e[1][i];
	d[0]=INF, vis[1]=1;
	for(int now,i=1; i<n; ++i)
	{
		now=0;
		for(int j=2; j<=n; ++j)
			if(!vis[j] && d[j]<d[now]) now=j;
		vis[now]=1, res+=d[now];
		for(int j=2; j<=n; ++j)
			if(!vis[j] && d[j]>e[now][j])
				d[j]=e[now][j];
	}
	return res<=0;
}
int main()
{
	while(n=read(),n)
	{
		for(int i=1; i<=n; ++i) x[i]=read(),y[i]=read(),z[i]=read();
		for(int i=1; i<n; ++i)
			for(int j=i+1; j<=n; ++j)
				dis[i][j]=Calc(i,j),cost[i][j]=std::abs(z[i]-z[j]);
		double l=0.0,r=101.0,mid;//r=多少啊。。
		while(r-l>=eps)
		{
			if(Check(mid=(l+r)/2.0)) r=mid;
			else l=mid;
		}
		printf("%.3f\n",l);//POJ不能用%lf! 惊了 刚知道
	}
	return 0;
}

2.Dinkelbach迭代

/*
20076K 297MS
并不明白原理 先将就用
*/
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
const int N=1005;
const double eps=1e-4,INF=1e8;
int n,x[N],y[N],z[N],cost[N][N],pre[N];
double dis[N][N],e[N][N],d[N];
bool vis[N];
inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
inline double Calc(int i,int j) {return sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+1.0*(y[i]-y[j])*(y[i]-y[j]));}
double Check(double x)
{
	for(int i=1; i<=n; ++i)
		for(int j=i+1; j<=n; ++j)
			e[j][i]=e[i][j]=1.0*cost[i][j]-x*dis[i][j];
	double Dis=0,Cost=0;//Prim
	memset(vis,0,sizeof vis);
	for(int i=2; i<=n; ++i) d[i]=e[1][i],pre[i]=1;
	d[0]=INF, vis[1]=1;
	for(int now,i=1; i<n; ++i)
	{
		now=0;
		for(int j=2; j<=n; ++j)
			if(!vis[j] && d[j]<d[now]) now=j;
		vis[now]=1, Dis+=dis[pre[now]][now], Cost+=cost[pre[now]][now];
		for(int j=2; j<=n; ++j)
			if(!vis[j] && d[j]>e[now][j])
				d[j]=e[now][j], pre[j]=now;
	}
	return Cost/Dis;
}
int main()
{
	while(n=read(),n)
	{
		for(int i=1; i<=n; ++i) x[i]=read(),y[i]=read(),z[i]=read();
		for(int i=1; i<n; ++i)
			for(int j=i+1; j<=n; ++j)
				dis[j][i]=dis[i][j]=Calc(i,j), cost[j][i]=cost[i][j]=std::abs(z[i]-z[j]);
		double x=0,y;
		while(1)
		{
			y=Check(x);
			if(fabs(x-y)<eps) break;
			x=y;
		}
		printf("%.3f\n",x);
	}
	return 0;
}