[BZOJ4827][Hnoi2017]礼物-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

【样例解释】

需要将第一个手环的亮度增加1，第一个手环的亮度变为： 2 3 4 5 6 旋转一下第二个手环。对于该样例，是将第二个手环的亮度6 3 3 4 5向左循环移动 2017-04-15 第 6 页，共 6 页 一个位置，使得第二手环的最终的亮度为:3 3 4 5 6。 此时两个手环的亮度差异值为1。

---

FFT裸题???
本蒟蒻完全没看出来……
连70分暴力都没想到……
显然是题做少了缺乏感觉……
太弱了……

(╯‵□′)╯︵┻━┻

---

思路:
事实上，我们要求的是
$\sum_{i=1}^n({x_i-y_{i+d}+c})^2$

那么展开得
$\sum_{i=1}^nx_i^2+y_{i+d}^2-2x_iy_{i+d}-2(x_i+y_{i+d})c+c^2$

发现其中只有 $2x_iy_{i+d}$ 无法较快预处理出来。
然后把x数组翻转，画一画x和y卷积的结果,你会发现卷积结果的第d项和第d+n项的和正是 $\sum_{i=1}^n2x_iy_{i+d}$ 的结果……

然后?没有然后了……

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

using namespace std;

typedef double ld;

const int N=500233;
const ld pi=acos(-1);

inline int read()
{
    int x=0;char ch=getchar();
    while(ch<'0' || '9'<ch)ch=getchar();
    while('0'<=ch && ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x;
}

int n,m,x[N],y[N],rev[N<<1],MMH,sx,sy,d,ma,len;
inline int maxx(int a,int b){if(a>b)return a;return b;}

struct vec
{
    ld x,y;
    vec(ld a=0.0,ld b=0.0)
    {
        x=a;y=b;
    }
}A[N],B[N],C[N];

vec operator + (vec satori,vec koishi)
{
    return vec(satori.x+koishi.x,satori.y+koishi.y);
}

vec operator - (vec satori,vec koishi)
{
    return vec(satori.x-koishi.x,satori.y-koishi.y);
}

vec operator * (vec satori,vec koishi)
{
    return vec(satori.x*koishi.x-satori.y*koishi.y,satori.x*koishi.y+koishi.x*satori.y);
}

void FFT(vec *a,int f)
{
    for(int i=0,j=0;i<len;i++)
    {
        if(i>j)
            swap(a[i],a[j]);
        for(int k=len>>1;(j^=k)<k;k>>=1);
    }

    for(int i=2;i<=len;i<<=1)
        for(int m=i>>1,j=0;j<len;j+=i)
        {
            vec wn=vec(cos(2.0*pi/i),sin(2.0*pi/i)*f);
            vec w=vec(1,0);
            for(int k=0;k<m;k++)
            {
                vec z=a[j+k+m]*w;
                a[j+k+m]=a[j+k]-z;
                a[j+k]=a[j+k]+z;
                w=w*wn;
            }
        }
}

int main()
{
    n=read();
    m=read();

    for(len=1;len<(n<<1);len<<=1);

    for(int i=0;i<n;i++)
        sx+=(x[i]=read());
    for(int i=0;i<n;i++)
        sy+=(y[i]=read());

    d=int(round(-1.0*(sx-sy)/n));
    for(int i=0;i<n;i++)
        MMH+=(x[i]+d)*(x[i]+d)+(y[i]*y[i]);
    for(int i=0;i<n;i++)
        A[i]=vec(x[n-i-1]+d),B[i]=vec(y[i]);

    FFT(A,1);FFT(B,1);
    for(int i=0;i<len;i++)
        C[i]=A[i]*B[i];
    FFT(C,-1);

    for(int i=0;i<n;i++)
        ma=maxx(ma,round((C[i].x+C[i+n].x)/len));

    printf("%d\n",MMH-ma-ma);

    return 0;
}