Codeforces429 D. Tricky Function（平面最近点对）

题意：

给一个长度为n的数组a
定义f(i，j)=(i-j)²+(a[i+1]+a[i+2]…a[j])²，其中i!=j
现在要你计算出f(i，j)的最小值

数据范围：n<=1e5

解法：

观察到f(i,j)中的第二项a[i+1]+d[i+2]..a[j]可以变为sum[j]-sum[i]
那么f(i,j)=(i-j)^2+(sum[j]-sum[i])^2
很像两点之间的距离式子,可以转化为(i,sum[i])与(j,sum[j])的距离
那么这题求最小的f(i,j)其实就是求所有点(i,sum[i])之间的最小距离
然后就变成了一道求平面最近点对的模板题了

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+5;
struct Point{//点
    int x,y;
}p[N],temp[N];
int a[N],n;
int sum[N];
bool cmpy(Point a,Point b){
    return a.y<b.y;
}
int getd(Point a,Point b){
    return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);
}
int cal(int l,int r){
    int d=1e18;
    if(l==r)return d;
    if(l+1==r)return getd(p[l],p[r]);
    int mid=(l+r)/2;
    int d1=cal(l,mid),d2=cal(mid+1,r);
    d=min(d1,d2);
    int k=0;
    for(int i=l;i<=r;i++){
        if(abs(p[mid].x-p[i].x)<=d)temp[++k]=p[i];
    }
    sort(temp+1,temp+k+1,cmpy);
    for(int i=1;i<=k;i++){
        for(int j=i+1;j<=k&&j<=i+6;j++){//temp[j].y-temp[i].y<d会超时,j<=i+6才能过,好像是数据卡了差值
            d=min(d,getd(temp[i],temp[j]));
        }
    }
    return d;
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[i];
    for(int i=1;i<=n;i++)p[i]={i,sum[i]};//x已经是升序了,不用排序
    int ans=cal(1,n);
    cout<<ans<<endl;
    return 0;
}

---