BZOJ4827: [Hnoi2017]礼物

题意大致是求这样一个柿子的最小值$\sum_{i=1}^n(A_i-B_j+k)^2$其中k是你定的，j是经过旋转后i的对应位置
然后画一下柿子
$\sum_{i=1}^n((A_i-B_j)^2+2k(A_i-B_j)+k^2)$
$\sum_{i=1}^n(A_i-B_j)^2+2k(\sum_{i=1}^nA_i-\sum_{i=1}^nB_i)+nk^2$
前面的柿子没有k，说明其实k的取值和i、j的对应关系就没什么关系了，他只和A，B的和相关
因为$A_i,B_i<=100$，所以$|k|<=100$，枚举找出后面的柿子的最小值即可
对于前面的柿子，继续画
$\sum_{i=1}^n(A_i-B_j)^2$
$\sum_{i=1}^n(A_i^2+B_j^2-2A_iB_j)$
$\sum_{i=1}^nA_i^2+\sum_{i=1}^nB_i^2-2\sum_{i=1}^nA_iB_j$
前面那个柿子直接求
然后我们想知道对于所有旋转后的对应关系，$\sum_{i=1}^nA_iB_j$的最大值
一种旋转后的对应关系，其实就是$i-j=a或b$的关系
比如下图

$i-j=-1或3$
而对于$i-j=a，\sum_{i=1}^nA_iB_j$可以通过卷积求
具体说，就是将$A,B$从0到n-1标号，然后将$B$数组翻转$B_j=B_{n-j-1}$
那么对于$X=i+n-j-1$，$X位置的卷积值=\sum_{i=0}^{n-1}A_iB_j(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;
}