编程之美初赛C题总结

描述

A市是一个高度规划的城市，但是科技高端发达的地方，居民们也不能忘记运动和锻炼，因此城市规划局在设计A市的时候也要考虑为居民们建造一个活动中心，方便居住在A市的居民们能随时开展运动，锻炼强健的身心。

城市规划局希望活动中心的位置满足以下条件：

1. 到所有居住地的总距离最小。

2. 为了方便活动中心的资源补给和其他器材的维护，活动中心必须建设在A市的主干道上。

为了简化问题，我们将A市摆在二维平面上，城市的主干道看作直角坐标系平的X轴，城市中所有的居住地都可以看成二维平面上的一个点。

现在，A市的城市规划局希望知道活动中心建在哪儿最好。

输入

第一行包括一个数T，表示数据的组数。

接下来包含T组数据，每组数据的第一行包括一个整数N，表示A市共有N处居住地

接下来N行表示每处居住地的坐标。

输出

对于每组数据，输出一行“Case X: Y”，其中X表示每组数据的编号(从1开始)，Y表示活动中心的最优建造位置。我们建议你的输出保留Y到小数点后6位或以上，任何与标准答案的绝对误差或者相对误差在10-6以内的结果都将被视为正确。

数据范围

小数据：1 ≤ T ≤ 1000, 1 ≤ N ≤ 10

大数据：1 ≤ T ≤ 10, 1 ≤ N ≤ 105

对于所有数据，坐标值都是整数且绝对值都不超过106

样例解释

样例1：活动中心的最优建造位置为(1.678787, 0)

样例输入

1
3
1 1
2 2
3 3

样例输出

Case 1: 1.678787

稍稍写了写公式，除了发现似乎是在求一个势能最低点以外，没什么头绪。

稍稍百度了一下之后被告知这似乎是一个NP问题，于是看数据量不大，妥妥地写了一发近似算法.......

赛后和人一聊，发现距离和的这个函数是单调的，于是证实了自己算法的正确性。

只可惜..........最后时刻编译器选错了.........世界的恶意......

不多说，牛顿迭代法逼近，比二分要快上N多倍的算法......听闻三分也能过，之后研究研究= =

距离之和是关于x轴上点P(x,0)的一个函数，求导，得到该函数的导函数是一个单调的函数，于是单零点可证，用牛顿迭代法求其零点，即为最小值。

迭代公式：

终止条件为相邻两次迭代结果相差不超过1e-7

为进一步加速算法，初始值取所有点横坐标的均值。

代码：

#include<iostream>
#include<cmath>
#include<cstdio>
#include<iomanip>
#define MAX 100001
using namespace std;

typedef struct{
    long long x;
    long long y;
}Point;
Point p[MAX];

double ave(int n){
    double sum=0.0;
    for(int i=0;i<n;i++)
       sum+=p[i].x;
    return sum/n;
}

void CalcSum(int n,long double &x){
    long double sum1=0.0,sum2=0.0;
    for(int i=0;i<n;i++){
        sum1+=(x-p[i].x)/sqrt((x-p[i].x)*(x-p[i].x)+p[i].y*p[i].y);
        sum2+=(p[i].y*p[i].y)/pow((x-p[i].x)*(x-p[i].x)+p[i].y*p[i].y,1.5);
    }
    x-=sum1/sum2;
}

int main(){
     int T,n;
     scanf("%d",&T);
     for(int i=1;i<=T;i++){
         scanf("%d",&n);
         for(int j=0;j<n;j++)
             scanf("%lld%lld",&p[j].x,&p[j].y);
         long double curr=ave(n),prev;
         do{
            prev=curr;
            CalcSum(n,curr);
         }while(abs(prev-curr)>1e-7);
         cout<<"Case "<<i<<": "<<setprecision(6)<<fixed<<curr<<endl;
     }
}