[bzoj4827][FFT]礼物

Description

我的室友最近喜欢上了一个可爱的小女生。马上就要到她的生日了，他决定买一对情侣手 环，一个留给自己，一 个送给她。每个手环上各有 n
个装饰物，并且每个装饰物都有一定的亮度。但是在她生日的前一天，我的室友突
然发现他好像拿错了一个手环，而且已经没时间去更换它了！他只能使用一种特殊的方法，将其中一个手环中所有 装饰物的亮度增加一个相同的自然数
c（即非负整数）。并且由于这个手环是一个圆，可以以任意的角度旋转它， 但是由于上面
装饰物的方向是固定的，所以手环不能翻转。需要在经过亮度改造和旋转之后，使得两个手环的差
异值最小。在将两个手环旋转且装饰物对齐了之后，从对齐的某个位置开始逆时针方向对装饰物编号 1,2,…,n， 其中 n
为每个手环的装饰物个数，第 1 个手环的 i 号位置装饰物亮度为 xi，第 2 个手 环的 i 号位置装饰物 亮度为
yi，两个手环之间的差异值为(参见输入输出样例和样例解释)： \sum_{i=1}^{n}(x_i-y_i)^2麻烦你帮他
计算一下，进行调整（亮度改造和旋转），使得两个手环之间的差异值最小， 这个最小值是多少呢？

Input

输入数据的第一行有两个数n, m，代表每条手环的装饰物的数量为n，每个装饰物的初始 亮度小于等于m。
接下来两行，每行各有n个数，分别代表第一条手环和第二条手环上从某个位置开始逆时 针方向上各装饰物的亮度。 1≤n≤50000,
1≤m≤100, 1≤ai≤m

Output

输出一个数，表示两个手环能产生的最小差异值。 注意在将手环改造之后，装饰物的亮度 可以大于 m。

Sample Input

5 6

1 2 3 4 5

6 3 3 4 5

Sample Output

1

HINT

【样例解释】

需要将第一个手环的亮度增加1，第一个手环的亮度变为： 2 3 4 5 6 旋转一下第二个手环。对于该样例，是将第

二个手环的亮度6 3 3 4 5向左循环移动 2017-04-15 第 6 页，共 6 页 一个位置，使得第二手环的最终的亮度为

：3 3 4 5 6。 此时两个手环的亮度差异值为1。

题解

省选复习算法系列
首先式子可以变成$sigma(xi+c-yi)$
大力推一下式子，原式变为:
$sigma((xi-yi)^2+c^2-2*(xi-yi)*c)$
设sumx表示第一个手环的前缀和，sumy表示第二个手环的前缀和，然后化成
$n*c^2-2*c(sumx-sumy)+sigma(xi^2)+sigma(yi^2)-2*sigma(xi*yi)$
前面的式子都是定值，关于c的式子是一个一元二次函数，直接用顶点坐标求一个最值。最后的$2*sigma(xi*yi)$，把y数组倒序，可以化成$sigma(x[i]*y[n-i+1])$，是一个卷积形式
由于可以旋转手环，可以想到这是一个循环卷积，我们把xi倍增再与yi相乘，最后枚举每一个可能的答案位置求解
卷积用fft优化一下即可

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=210000;
const double PI=acos(-1.0);
struct Complex
{
    double r,i;
    Complex(){}
    Complex(double _r,double _i){r=_r;i=_i;}
    friend Complex operator +(const Complex &x,const Complex &y){return Complex(x.r+y.r,x.i+y.i);}
    friend Complex operator -(const Complex &x,const Complex &y){return Complex(x.r-y.r,x.i-y.i);}
    friend Complex operator *(const Complex &x,const Complex &y){return Complex(x.r*y.r-x.i*y.i,x.r*y.i+x.i*y.r);} 
}a[MAXN],b[MAXN];
int R[MAXN],L;
void fft(Complex *y,int len,int on)
{
    for(int i=0;i<len;i++)if(i<R[i])swap(y[i],y[R[i]]);
    for(int i=1;i<len;i<<=1)
    {
        Complex wn(cos(PI/i),sin(on*PI/i));
        for(int j=0;j<len;j+=(i<<1))
        {
            Complex w(1,0);
            for(int k=0;k<i;k++)
            {
                Complex u=y[j+k];
                Complex v=y[j+k+i]*w;
                y[j+k]=u+v;
                y[j+k+i]=u-v;
                w=w*wn;
            }
        }
    }
    if(on==-1)for(int i=0;i<len;i++)y[i].r/=len;
}
int n,m,len;
double c,tt,sumx,sumy;
int main()
{
    scanf("%d%d",&n,&m);
    L=0;
    for(len=1;len<=n*2;len<<=1)L++;
    for(int i=0;i<len;i++)R[i]=(R[i>>1]>>1)|(i&1)<<(L-1);
    for(int i=0;i<n;i++)
    {
        int u;scanf("%d",&u);
        a[i]=a[i+n]=Complex(u,0);
        sumx+=u*u;tt-=u;
    }
    for(int i=0;i<n;i++)
    {
        int u;scanf("%d",&u);
        b[n-i-1]=Complex(u,0);
        sumy+=u*u;tt+=u;
    }
    c=round(tt/n);
    fft(a,len,1);fft(b,len,1);
    for(int i=0;i<len;i++)a[i]=a[i]*b[i];
    fft(a,len,-1);
    double ans=0x3f3f3f3f;
    for(int i=n;i<len;i++)ans=min(ans,n*c*c-2*c*tt+sumx+sumy-2*(a[i].r+0.1));
    printf("%.0lf\n",ans);
    return 0;
}