题意大致是求这样一个柿子的最小值
∑ni=1(Ai−Bj+k)2
其中k是你定的,j是经过旋转后i的对应位置
然后画一下柿子
∑ni=1((Ai−Bj)2+2k(Ai−Bj)+k2)
∑ni=1(Ai−Bj)2+2k(∑ni=1Ai−∑ni=1Bi)+nk2
前面的柿子没有k,说明其实k的取值和i、j的对应关系就没什么关系了,他只和A,B的和相关
因为
Ai,Bi<=100
,所以
|k|<=100
,枚举找出后面的柿子的最小值即可
对于前面的柿子,继续画
∑ni=1(Ai−Bj)2
∑ni=1(A2i+B2j−2AiBj)
∑ni=1A2i+∑ni=1B2i−2∑ni=1AiBj
前面那个柿子直接求
然后我们想知道对于所有旋转后的对应关系,
∑ni=1AiBj
的最大值
一种旋转后的对应关系,其实就是
i−j=a或b
的关系
比如下图
i−j=−1或3
而对于
i−j=a,∑ni=1AiBj
可以通过卷积求
具体说,就是将
A,B
从0到n-1标号,然后将
B
数组翻转
那么对于
X=i+n−j−1
,
X位置的卷积值=∑n−1i=0AiBj(i−j=X−n+1)
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 410000;
const double pi = acos(-1);
int n,N;
struct E
{
double x,y;
E(){}
E(const double _x,const double _y){x=_x;y=_y;}
friend E operator +(E x,E y){return E(x.x+y.x,x.y+y.y);}
friend E operator -(E x,E y){return E(x.x-y.x,x.y-y.y);}
friend E operator *(E x,E y)
{
return E(x.x*y.x-x.y*y.y,x.x*y.y+x.y*y.x);
}
}w[maxn],f1[maxn],f2[maxn]; int id[maxn];
int a[maxn],b[maxn];
void DFT(E f[],const int sig)
{
for(int i=0;i<N;i++) if(i<id[i]) swap(f[i],f[id[i]]);
for(int m=2;m<=N;m<<=1)
{
int t=m>>1,tt=N/m;
for(int i=0;i<t;i++)
{
E wn=sig==1?w[i*tt]:w[N-i*tt];
for(int j=i;j<N;j+=m)
{
E tx=f[j],ty=wn*f[j+t];
f[j]=tx+ty;
f[j+t]=tx-ty;
}
}
}
if(sig==-1) for(int i=0;i<N;i++) f[i].x/=N;
}
void pre()
{
N=1; int ln=0;
while(N<=(n<<1)) N<<=1,ln++;
for(int i=1;i<N;i++) id[i]=(id[i>>1]>>1)|((i&1)?1<<ln-1:0);
for(int m=2;m<=N;m<<=1)
{
int t=m>>1,tt=N/m;
for(int i=0;i<t;i++)
w[i*tt]=E(cos(2*pi*i/m),sin(2*pi*i/m)),
w[N-i*tt]=E(cos(2*pi*i/m),sin(-2*pi*i/m));
}
}
int r[maxn];
int main()
{
scanf("%d%*d",&n); pre();
for(int i=0;i<n;i++) scanf("%d",&a[i]);
for(int i=0;i<n;i++) scanf("%d",&b[i]);
ll ans=0;
int s1=0,s2=0;
for(int i=0;i<n;i++)
ans+=a[i]*a[i]+b[i]*b[i],s1+=a[i],s2+=b[i];
int s=s1-s2; int tt=INT_MAX;
for(int k=-100;k<=100;k++) tt=min(tt,2*k*s+n*k*k);
ans+=(ll)tt;
for(int i=0;i<n;i++) f1[i]=E(a[i],0),f2[n-i-1]=E(b[i],0);
DFT(f1,1); DFT(f2,1);
for(int i=0;i<N;i++) f1[i]=f1[i]*f2[i];
DFT(f1,-1);
for(int i=0;i<n+n-1;i++)
{
int k=i-n+1;
if(k<0) r[n+k]+=(int)(f1[i].x+0.5);
else r[k]+=(int)(f1[i].x+0.5);
}
tt=INT_MIN;
for(int i=0;i<n;i++) tt=max(tt,r[i]);
ans-=2ll*tt;
printf("%lld\n",ans);
return 0;
}