BZOJ4827:[HNOI2017]礼物(FFT)

大家好，又见面了，我是你们的朋友全栈君。

Description

我的室友最近喜欢上了一个可爱的小女生。马上就要到她的生日了，他决定买一对情侣手环，一个留给自己，一

个送给她。每个手环上各有 n 个装饰物，并且每个装饰物都有一定的亮度。但是在她生日的前一天，我的室友突

然发现他好像拿错了一个手环，而且已经没时间去更换它了！他只能使用一种特殊的方法，将其中一个手环中所有

装饰物的亮度增加一个相同的自然数 c（即非负整数）。并且由于这个手环是一个圆，可以以任意的角度旋转它，

但是由于上面装饰物的方向是固定的，所以手环不能翻转。需要在经过亮度改造和旋转之后，使得两个手环的差

异值最小。在将两个手环旋转且装饰物对齐了之后，从对齐的某个位置开始逆时针方向对装饰物编号 1,2,…,n，

其中 n 为每个手环的装饰物个数，第 1 个手环的 i 号位置装饰物亮度为 xi，第 2 个手环的 i 号位置装饰物

亮度为 yi，两个手环之间的差异值为(参见输入输出样例和样例解释)：
BZOJ4827:[HNOI2017]礼物(FFT)

麻烦你帮他

计算一下，进行调整（亮度改造和旋转），使得两个手环之间的差异值最小，这个最小值是多少呢？

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。

Solution

感谢XY大爷对我的帮助

我以后FFT下标再也不从1开始了太难处理了QAQ

BZOJ4827:[HNOI2017]礼物(FFT)

Code

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #include<cmath>
 5 #define N (600000+100)
 6 using namespace std;
 7 
 8 double pi=acos(-1.0),F[N];
 9 int n,m,k,fn,l,r[N],A[N],B[N];
10 int minn=0x7fffffff,squA,squB,squC,sumA,sumB;
11 struct complex
12 {
13     double x,y;
14     complex (double xx=0,double yy=0)
15     {
16         x=xx; y=yy;
17     }
18 }a[N],b[N];
19 
20 complex operator + (complex a,complex b){
   
   return complex(a.x+b.x,a.y+b.y);}
21 complex operator - (complex a,complex b){
   
   return complex(a.x-b.x,a.y-b.y);}
22 complex operator * (complex a,complex b){
   
   return complex(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
23 complex operator / (complex a,double b){
   
   return complex(a.x/b,a.y/b);}
24 
25 void FFT(int n,complex *a,int opt)
26 {
27     for (int i=0; i<n; ++i)
28         if (i<r[i])
29             swap(a[i],a[r[i]]);
30     for (int k=1; k<n; k<<=1)
31     {
32         complex wn=complex(cos(pi/k),opt*sin(pi/k));
33         for (int i=0; i<n; i+=k<<1)
34         {
35             complex w=complex(1,0);
36             for (int j=0; j<k; ++j,w=w*wn)
37             {
38                 complex x=a[i+j], y=w*a[i+j+k];
39                 a[i+j]=x+y; a[i+j+k]=x-y;
40             }
41         }
42     }
43     if (opt==-1) for (int i=0; i<n; ++i) a[i]=a[i]/n;
44 }
45 
46 int Calc()
47 {
48     for (int i=0; i<n; ++i) a[i].x=A[i];
49     for (int i=0; i<m; ++i) b[i].x=B[i];
50     
51     FFT(fn,a,1); FFT(fn,b,1);
52     for (int i=0; i<=fn; ++i) a[i]=a[i]*b[i];
53     FFT(fn,a,-1);
54     
55     for (int i=0; i<n; ++i) squA=squA+A[i]*A[i];
56     for (int i=0; i<n; ++i) squB=squB+B[i]*B[i];
57     for (int i=0; i<n; ++i) sumA=sumA+A[i];
58     for (int i=0; i<n; ++i) sumB=sumB+B[i];
59     
60     for (int c=-k; c<=k; ++c)
61         for (int i=n-1; i<=m-1; ++i)
62             minn=min(minn,n*c*c+squA+2*c*sumA+squB-2*(int)(a[i].x+0.5)-2*c*sumB);
63     return minn;
64 }
65 
66 int main()
67 {
68     scanf("%d%d",&n,&k);
69     for (int i=0; i<n; ++i) scanf("%d",&A[n-i-1]);
70     for (int i=0; i<n; ++i) scanf("%d",&B[i]),B[n+i]=B[i];
71     m=2*n;
72     
73     fn=1;
74     while (fn<=n+m) fn<<=1, l++;
75     for (int i=0; i<fn; ++i)
76         r[i]=(r[i>>1]>>1) | ((i&1)<<(l-1));
77     
78     printf("%d\n",Calc());
79 }