首先这是要解决什么问题:一个带权完全图,每条边都有自己的花费值cost[i]和收益值benifit[i],如果用x[i]来代表一条边取或不取,那么求一个生成树。要求:r=(∑cost[i]*x[i] ) / (∑benifit[i]*x[i] )最小。
经典题目:POJ2728 - Desert King
如何来求解:这里用到了0-1分数规划思想,对于上式可以变形为 z(r)=∑cost[i]*x[i] -r*∑benifit[i]*x[i]。而z(r)=0为我们所求。
这里有个非常重要的结论:z(r)为单调递减函数,因此是线性的。于是"我们可以兴高采烈地把z(r)看做以 cost[i]-r*benifit[i] 为边权的最小生成树的总权值"。显然,我们所求即为z(max(r))=0。这里max(r)是由z=0确定的。
于是有了两种算法:
1、Dinkelbanch算法,即迭代法。取r的一个初值,一般为0,得到一个生成树之后,令r=(∑benifit[i]*x[i] ) / (∑cost[i]*x[i] ),哐哐迭代之后……r是趋于最大值的,而z趋于0。
2、二分法,z(r)=0是我们所求,而它又是单调函数,那么二分r即可。据说二分比迭代慢了很多。
P.S.:最大密度子图据说也可以用这个算法解决,恩,又听到一个新名词……
代码:
#include
#include
#include
#include
#include
#include
using namespace std;
const int MAXN = 1100;
const int INF = 0x3f3f3f3f ;
double grap[MAXN][MAXN] , xy[MAXN][MAXN];
int n , bian[MAXN][3] , cost[MAXN][MAXN];
double solve();
void add()
{
int i , j;
for(i = 1; i <= n; i++)
for(j = i+1 ; j <= n; j++)
{
int g = bian[i][0] - bian[j][0] , h = bian[i][1] - bian[j][1];
grap[i][j] = grap[j][i] = sqrt(g*g*1.0 + h*h*1.0);
cost[i][j] = cost[j][i] = abs(bian[i][2] - bian[j][2]);
}
printf("%.3lf\n" , solve());
}
void makmap(double x)
{
for(int i = 1; i <= n; i++)
for(int j = i+1 ; j <= n; j++)
{
xy[i][j] = xy[j][i] = cost[i][j] - x*grap[i][j];
}
}
double prim()
{
int i , u = 1 , done[MAXN];
double d[MAXN] , sum = 0.0;
for(i = 1; i <= n; i++) d[i] = 100000000.0;
memset(done , 0 , sizeof(done));
done[1] = 1;
for(; ; )
{
double maxx = 100000000.0 ;
int x = -1;
for(i = 2; i <= n; i++)
{
if(!done[i] && d[i] > xy[u][i])
d[i] = xy[u][i];
if(!done[i] && maxx > d[i])
{
x = i;
maxx = d[i];
}
}
if(x == -1) break;
sum += maxx;
u = x;
done[u] = 1;
}
//cout<<sum<<endl;
return sum;
}
double solve()
{
double l = 0.0 , r = 1000000000.0;
while((r-l) > 0.0001)
{
double m = (r+l)/2;
makmap(m);
double tem = prim();
if(0 <= tem && tem< 0.00001) return m;
if(tem >= 0) l = m;
else r = m;
//cout<<r<<","<<l<<endl;
}
return l;
}
int main()
{
while(scanf("%d" , &n) && n)
{
int i ;
for(i = 1; i <= n; i++)
scanf("%d %d %d" , &bian[i][0] , &bian[i][1] , &bian[i][2]);
add();
}
return 0;
}