自从htn体验了一把飙车的快感,他就下定决心要闭门造车!但是他两手空空怎么造得出车来呢?无奈的他只好来到了汽车零部件商店。
一走进商店,玲琅满目的各式零件看得htn眼花缭乱。但是他很快便反应过来:我只要买一套好的零件就行。首先它们的性能差不能太大,否则汽车的兼容性不好,开着开着就损坏了;其次,当然是越便宜越好了!为了打造一辆顶级跑车,htn陷入了沉思……
现在商店中有 N 件零件,给出这 N 件零件的价格,其性能等于价格。htn要从中购买一套零件,即选取这个序列的一个子串(连续一段)。要求如下:
1、这一套零件个数要大于等于2(这才算一套)。
2、这套零件的性能差为首尾两个零件的性能差(htn觉得每一个都比较性能差实在是太累了)。
3、购买这套零件的价格和为它们各自价格的总和。
4、最终的总花费为 性能差²+价格和²。
5、由于商店最近有优惠活动,所以每一套零件的第一个都是免费的。
我们发现,如果令s[i]=Σa[j]{0<j<=i} 那么,answer=min((a[i]-a[j])^2+(s[i]-s[j])^2)
看到两个平方想到了什么?
平面最近点对!
采用分治法,可以自行百度,注意一下细节:
我们在分治的时候,顺便进行归并排序,这样就可以避免调用sort,将复杂度降到O(n lg n)
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 100010
#define LL long long
using namespace std;
LL x[N],y[N],a[N],b[N]; int n;
inline LL sqr(LL x){ return x*x; }
inline LL dis(int i,int j){ return sqr(x[i]-x[j])+sqr(y[i]-y[j]); }
LL merge(int l,int r){if(l==r) return 1ll<<62;int m=l+r>>1,c=l; LL d=min(merge(l,m),merge(m+1,r));for(int i=l,j=m+1;i<=m || j<=r;++i){for(;j<=r && (i>m||y[i]>y[j]);++j) a[c]=x[j],b[c++]=y[j];if(i<=m && abs(x[i]-x[m])<d)for(int k=max(m+1,j-3);k<=j+3;++k) d=min(d,dis(i,k));if(i<=m) a[c]=x[i],b[c++]=y[i];}for(int i=l;i<=r;++i) x[i]=a[i],y[i]=b[i];return d;
}
int main(){scanf("%d",&n);for(int i=1;i<=n;++i) scanf("%lld",y+i),x[i]=y[i]+x[i-1];printf("%lld\n",merge(1,n));
}