bzoj4827: [Hnoi2017]礼物 FFT

bzoj4827: [Hnoi2017]礼物

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。

分析

考虑一种旋转方案

$$ans=\sum (x_i-y_i+c)^2$$
$$=\sum (x_i^2+y_i^2) -2\sum x_iy_i+nc^2-2c\sum x_i-y_i$$
令 $A=\sum (x_i^2+y_i^2),B=\sum x_i-y_i,A,B$不受旋转影响
原式等价于 $ans=A+nc^2-2Bc-2\sum x_iy_i$
所以c的取值和旋转无关，一个二次函数最小值可以搞定。
于是问题等价于 $Maximize(\sum x_{(i+c) \mod n}y_i)$
其中c是旋转的参数，令 $c \in [0,n-1]$。
循环卷积不舒服，先倍增环，于是得到目标函数
$Maximize(\sum x_{i+c}y_i)$
这个东西有一个景点的处理方法，就是把 $y_i$翻转，得到
$Maximize(\sum x_{i+c}y_{n-i-1})$
于是得到了一个卷积的形式，卷积完答案区间在 $[n-1,2n-1]$中，一个FFT贴上去然后枚举即可。

代码

#include<cstdio>
#include<algorithm>
#include<cmath>
const int N = 1 << 20;
const double pi = acos(-1.0);
int ri() {
    char ch = getchar(); int x = 0;
    for(;ch < '0' || ch > '9'; ch = getchar()) ;
    for(;ch >= '0' && ch <= '9'; ch = getchar()) x = (x << 1) + (x << 3) - '0' + ch;
    return x;
}
struct cp {
    double r, i;
    cp(double _r = 0, double _i = 0) : r(_r), i(_i) {}
    cp operator + (cp a) {return cp(r + a.r, i + a.i);}
    cp operator - (cp a) {return cp(r - a.r, i - a.i);}
    cp operator * (cp a) {return cp(r * a.r - i * a.i, r * a.i + i * a.r);}
}a[N], b[N], w[N];
int n, L, R[N], x[N], y[N], sum;
void Pre() {
    n = ri(); ri(); int sx, sy; sx = sy = 0;
    for(int i = 0;i < n; ++i) x[i] = ri(), sx += x[i], sum += x[i] * x[i];
    for(int i = 0;i < n; ++i) y[i] = ri(), sy += y[i], sum += y[i] * y[i];
    int c = round((double)(sy - sx) / (double)n); sum += n * c * c + ((sx - sy) * c << 1);
    for(int i = 0;i < n; ++i) b[i].r = y[n - i - 1];
    for(int i = 0;i < (n << 1); ++i) a[i].r = x[i % n];
    int x = 0; for(L = 1; (L <<= 1) <= (n * 3); ++x) ;
    for(int i = 1;i < L; ++i) R[i] = (R[i >> 1] >> 1) | ((i & 1) << x);
    for(int i = 0;i < L; ++i) w[i] = cp(cos(2 * pi * i / L), sin(2 * pi * i / L));
}
void FFT(cp *F) {
    for(int i = 0;i < L; ++i) if(i < R[i]) std::swap(F[R[i]], F[i]);
    for(int i = 1, d = L >> 1; i < L; i <<= 1, d >>= 1)
        for(int j = 0;j < L; j += (i << 1)) {
            cp *l = F + j, *r = F + j + i, *p = w, tp   ;
            for(int k = 0;k < i; ++k, ++l, ++r, p += d) 
                tp = *r * *p, *r = *l - tp, *l = *l + tp;
        }
}
int main() {
    Pre(); FFT(a); FFT(b);
    for(int i = 0;i < L; ++i) a[i] = a[i] * b[i];
    FFT(a); int ans = 0;
    for(int i = 1; i < (L >> 1); ++i) std::swap(a[i], a[L - i]);
    for(int i = n - 1;i < (n << 1) - 1; ++i) ans = std::max(ans, int(a[i].r / L + 0.1));
    printf("%d\n", sum - (ans << 1));
    return 0;
}