POJ 2728 Desert King(最优比率生成树, 01分数规划)

题意:

给定n个村子的坐标(x,y)和高度z, 求出修n-1条路连通所有村子, 并且让 修路花费/修路长度 最少的值

两个村子修一条路, 修路花费 = abs(高度差), 修路长度 = 欧氏距离

分析:

01分数划分的题目, 构造出 d[i] = 修路花费 - L * 修路长度, 这个L值我们可以二分(这道题看数据范围的话二分上限其实挺大的, 但其实上限取到100就可以过), 也可以用Dinkelbach迭代出来。

二分(1422ms)

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <cstring>
#include <cmath>
#define rep(i,a,b) for(int i = a; i < b;i++)
#define _rep(i,a,b) for(int i = a; i <= b;i++)
using namespace std;
const int maxn = + ;
const double inf = 1e9 + ;
const double eps = 1e-;
int n; double G[maxn][maxn];
double x[maxn], y[maxn], z[maxn]; inline double p2p_dis(double x1, double y1, double x2, double y2){
return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
double prim(double L){
double tree_dis = ;
double dis[maxn];
int vis[maxn];
fill(dis, dis + maxn, inf);
memset(vis, ,sizeof(vis)); dis[] = ;
vis[] = ; for(int i = ; i < n; i++) dis[i] = abs(z[] - z[i]) - L * G[][i]; for(int times = ; times < n - ; times++){
int picked = -;
double min_dis = inf;
for(int i = ; i < n; i++){
if(!vis[i] && dis[i] < min_dis){
min_dis = dis[i];
picked = i;
}
}
tree_dis += min_dis;
vis[picked] = ;
for(int i = ; i < n; i++)
if(!vis[i])
dis[i] = min(dis[i] , abs(z[picked] - z[i]) - L * G[picked][i]) ;
}
return tree_dis;
} int main(){
// freopen("1.txt","r", stdin);
while(cin >> n && n){
rep(i,,n) cin >> x[i] >> y[i] >> z[i];
rep(i,,n) rep(j,i+,n) G[i][j] = G[j][i] = p2p_dis(x[i],y[i],x[j],y[j]);
double l = , r = ;
while(abs(r - l) > eps){
double mid = (l + r) / ;
// printf("%.3f\n", mid);
if(prim(mid) >= ){
l = mid;
}else{
r = mid;
}
}
printf("%.3f\n", l);
}
return ;
}

Dinkelbach(204ms)

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <cstring>
#include <cmath>
#define rep(i,a,b) for(int i = a; i < b;i++)
#define _rep(i,a,b) for(int i = a; i <= b;i++)
using namespace std;
const int maxn = + ;
const double inf = 1e9 + ;
const double eps = 1e-;
int n; double G[maxn][maxn];
double x[maxn], y[maxn], z[maxn]; inline double p2p_dis(double x1, double y1, double x2, double y2){
return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
double prim(double L){
double ele = , deno = ; //分子分母
double dis[maxn];
int near[maxn]; //开一个数组记录每次选中是哪一条边 int vis[maxn];
fill(dis, dis + maxn, inf);
memset(vis, ,sizeof(vis));
memset(near, -, sizeof(near));
dis[] = ;
vis[] = ; for(int i = ; i < n; i++) dis[i] = abs(z[] - z[i]) - L * G[][i], near[i] = ; //每个点都是由0点更新的, 所以near都是0 for(int times = ; times < n - ; times++){
int picked = -;
double min_dis = inf;
for(int i = ; i < n; i++){
if(!vis[i] && dis[i] < min_dis){
min_dis = dis[i];
picked = i;
}
}
ele += abs(z[near[picked]] - z[picked]);//分子是选的路程
deno += G[near[picked]][picked]; //分母是真实的花费
vis[picked] = ;
for(int i = ; i < n; i++)
if(!vis[i] && dis[i] > abs(z[picked] - z[i]) - L * G[picked][i]){
dis[i] = abs(z[picked] - z[i]) - L * G[picked][i];
near[i] = picked;//如果通过picked点更新dis,那么该位置near就是picked
} } return ele/deno; //注意返回的跟二分不一样, 应该返回当前L下真正题目的所求
} int main(){
// freopen("1.txt","r", stdin);
while(cin >> n && n){
rep(i,,n) cin >> x[i] >> y[i] >> z[i];
rep(i,,n) rep(j,i+,n) G[i][j] = G[j][i] = p2p_dis(x[i],y[i],x[j],y[j]);
double L, ans = ;
while(){
L = ans;
ans = prim(L); //这是一个真实值, 不是含L的变式
if(abs(ans - L)< eps) break;
}
printf("%.3f\n", ans);
}
return ;
}
上一篇:POJ2728 Desert King 最优比率生成树


下一篇:advance 模板 怎么生成module