这一题第一眼看到就感觉很像三分搜索。先考虑一个简单的版本,加入center的X坐标固定,那么Y坐标从1000到-1000移动,对于某一个点(Xi,Yi),对应的cost fi(Y)=max(|X-Xi|,|Y-Yi|)*Wi是先减小,到达max(|X-Xi|,|Y-Yi|)=|X-Xi|后变成一个常数,再增大。是一个凹函数。对于每个点的变化规律都是如此,那么所有点的cost加起来也是凹函数。intuitive的解释就是,对于每个点,fi(Y)的二阶导<0,那么加起来的函数的二阶导也是<0。
下面考虑将X从-1000到+1000移动的情形。记Y(X)为固定横坐标为X时,对应的竖线上的cost最小的点的纵坐标,考虑(X,Y(X))从左往右移动,对于某一个点(Xi,Yi),对应的cost也是先减小后增大,也是凹函数。
直观的解释是,对于X'<X (slightly smaller than),|X'-Xi|>|X-Xi|。所以在X'处,Y从大变小过程中达到的常数区间对应的cost比在X处更大。对于X'>X同理。Note that对于一个点的case,center选在这个点的位置上是最优的。
所以就可以三分嵌套求解。外层的三分搜索X,给定某个X之后,再用内层的三分搜索Y(X),就可以得出选择当前X时对应的cost的最小值。
#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<string>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
//Kickstart 2017 Round B Problem B. Center
int T;
const int maxn=10010;
int N;
double maxrange=1000;
double minrange=-1000;
double W[maxn];
double X[maxn];
double Y[maxn];
double ans;
const double eps=1e-9;
double calweight(double x,double y)
{
double ret=0;
for(int i=0;i<N;i++)
{
ret+=max(abs(x-X[i]),abs(y-Y[i]))*W[i];
}
return ret;
}
double trisearchinner(double x)//search along y axis
{
double low=minrange;
double up=maxrange;
double mid=(low+up)/2;
double midmid=(mid+up)/2;
while(low+eps<up)
{
mid=(low+up)/2;
midmid=(mid+up)/2;
if(calweight(x,mid)<calweight(x,midmid))
{
up=midmid;
}
else
{
low=mid;
}
}
return calweight(x,low);
}
void trisearchouter()//search along x axis
{
double left=minrange;
double right=maxrange;
double mid=(left+right)/2;
double midmid=(mid+right)/2;
while(left+eps<right)
{
mid=(left+right)/2;
midmid=(mid+right)/2;
if(trisearchinner(mid)<trisearchinner(midmid))
{
right=midmid;
}
else
{
left=mid;
}
}
ans=trisearchinner(left);
}
int main()
{
freopen("B-large-practice.in","r",stdin);
// freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
scanf("%d",&T);
for(int ca=1;ca<=T;ca++)
{
memset(W,0,sizeof(W));
memset(X,0,sizeof(X));
memset(Y,0,sizeof(Y));
ans=0;
scanf("%d",&N);
//cout<<N<<endl;
for(int i=0;i<N;i++)
{
scanf("%lf %lf %lf",&X[i],&Y[i],&W[i]);
//cout<<X[i]<<" "<<Y[i]<<" "<<W[i]<<endl;
}
trisearchouter();
//cout<<ans<<endl;
printf("Case #%d: %.6f\n",ca,ans);
}
return 0;
}