Problem Address:http://poj.org/problem?id=2728
【前言】
我确实是一拿到手就把它当MST做了。
但是经历了一次TLE后我居然翻出算法导论硬生生地把Prim算法的复杂度变小了。
也才发现原来一直以来我写的Prim都是n^3的,而优化了的Prim是O(E+VlgV)。
嗯,确实是一个不小的收获。
但是发现Prim错了。
于是翻出了解题报告,看到了最优比率生成树。
嗯,又有收获了。
【思路】
先说说Prim。
以前写Prim就是三重循环,第一重循环处理n个点,第二重循环找未访问的点以处理,第三重循环找该点的最小值。
后两重循环实际是用来查找最小值的。
然后重温了一下算法导论。
第一重循环处理n个点,第二重由于有key[ ]和π[ ]维护,所以找最小值的时候是lg(n)【实际上我直接还是按n算】,更新时为n。
所以总复杂度为O(E+VlgV)。
事实上两者的不同是思路上的优化。
接下来说最优比率生成树。
实际上它是Prim+二分或迭代。
可以参考:http://www.cnblogs.com/lotus3x/archive/2009/03/21/1418480.html 和 http://blog.csdn.net/cicirise/article/details/3904362,里面都讲的很全面。
和上一篇文章类似,二分实际上就是找一个比率,然后看看在这个比率的作用下的结果,然后用过不断进行Prim运算,调整比率,直到满足条件。
而迭代则是每次Prim得出一个rate,然后不断进行Prim运算从而更新rate。
在这个题目中二分的次数比rate的次数高出很多,所以迭代的时间比较小。
除此之外,在代码方面,二分和迭代都是差不多的。
还有一点要注意的是,上述第一个链接是以c的总和作为二分的上限,但是在这道题中将其作为上限则会超时。这里定上限为1000则可通过。
【代码】
二分:
#include <iostream>
#include <cmath>
using namespace std;
const int maxn = 1000;
const double MAX = 99999999.00;
double map[maxn+5][maxn+5];
double c[maxn+5][maxn+5];
double b[maxn+5][maxn+5];
int x[maxn+5], y[maxn+5];
int z[maxn+5];
int visited[maxn+5];
double key[maxn+5];
int fa[maxn+5];
double MST(int n, double mid)
{
int ct = 1;
double temp;
int tb, tp;
int i, j;
for (i=0; i<n; i++)
visited[i] = 0;
for (i=0; i<n; i++)
{
map[i][i] = MAX;
for (j=i+1; j<n; j++)
{
map[i][j] = map[j][i] = c[i][j] - mid * b[i][j];
}
}
for (i=1; i<n; i++)
{
key[i] = map[0][i];
fa[i] = 0;
}
visited[0] = 1;
double sum = 0.0;
while(ct<n)
{
temp = MAX;
for (i=0; i<n; i++)
{
if (visited[i]==0 && key[i]<temp)
{
temp = key[i];
tb = fa[i];
tp = i;
}
}
visited[tp] = 1;
for (i=0; i<n; i++)
{
if (visited[i]==0 && map[tp][i]<key[i])
{
key[i] = map[tp][i];
fa[i] = tp;
}
}
ct++;
sum += temp;
}
return sum;
}
int main()
{
int n;
int i, j;
double high, mid, low;
double minb, t;
while(scanf("%d", &n)!=EOF)
{
if (n==0) break;
for (i=0; i<n; i++)
{
scanf("%d %d %d", &x[i], &y[i], &z[i]);
}
high = 0.0;
minb = MAX;
for (i=0; i<n; i++)
{
b[i][i] = c[i][i] = MAX;
for (j=i+1; j<n; j++)
{
c[i][j] = c[j][i] = (double)abs(z[i]-z[j]);
b[i][j] = b[j][i] = sqrt((double)((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])));
if (b[i][j]<minb)
minb = b[i][j];
// high += c[i][j];
}
}
// high /= minb;
high = 1000.0;
low = 0.0;
while(high>low)
{
mid = (high+low)/2;
t = MST(n, mid);
if (t>1e-4)
low = mid;
else if (t<-1e-4)
high = mid;
else
break;
}
printf("%.3lf\n", mid);
}
return 0;
}
迭代:
#include <iostream>
#include <cmath>
using namespace std;
const int maxn = 1000;
const double MAX = 99999999.00;
double map[maxn+5][maxn+5];
double c[maxn+5][maxn+5];
double b[maxn+5][maxn+5];
int x[maxn+5], y[maxn+5];
int z[maxn+5];
int visited[maxn+5];
double key[maxn+5];
int fa[maxn+5];
double MST(int n, double rate)
{
int ct = 1;
double temp;
int tb, tp;
int i, j;
for (i=0; i<n; i++)
visited[i] = 0;
for (i=0; i<n; i++)
{
map[i][i] = MAX;
for (j=i+1; j<n; j++)
{
map[i][j] = map[j][i] = c[i][j] - rate * b[i][j];
}
}
for (i=1; i<n; i++)
{
key[i] = map[0][i];
fa[i] = 0;
}
visited[0] = 1;
double tola = 0.0, tolb = 0.0;
while(ct<n)
{
temp = MAX;
for (i=0; i<n; i++)
{
if (visited[i]==0 && key[i]<temp)
{
temp = key[i];
tb = fa[i];
tp = i;
}
}
visited[tp] = 1;
for (i=0; i<n; i++)
{
if (visited[i]==0 && map[tp][i]<key[i])
{
key[i] = map[tp][i];
fa[i] = tp;
}
}
ct++;
tola += c[tb][tp];
tolb += b[tb][tp];
}
return tola/tolb;
}
int main()
{
int n;
int i, j;
double rate, cur;
while(scanf("%d", &n)!=EOF)
{
if (n==0) break;
for (i=0; i<n; i++)
{
scanf("%d %d %d", &x[i], &y[i], &z[i]);
}
for (i=0; i<n; i++)
{
b[i][i] = c[i][i] = MAX;
for (j=i+1; j<n; j++)
{
c[i][j] = c[j][i] = (double)abs(z[i]-z[j]);
b[i][j] = b[j][i] = sqrt((double)((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])));
}
}
rate = 0.0;
cur = MST(n, rate);
while(fabs(rate-cur)>1e-4)
{
rate = cur;
cur = MST(n, rate);
}
printf("%.3lf\n", cur);
}
return 0;
}
【P.S】
退出了数据挖掘。
放弃了软件杯比赛。
在ACM上好好加油!