题目描述
输入
输出
样例输入
5 6
1 2 3 4 5
6 3 3 4 5
样例输出
1
题解
FFT
首先,不用枚举c!
由于要求的是相对关系,所以给第二个手环+c就是给第一个手环-c。
设旋转后i位置分别为xi和yi,那么通过上面的式子可以得出c的最优取值与x和y的对应关系无关。
也就是说无论如何旋转,c的最优值总是固定的(sumy-sumx)/n(四舍五入到整数)
这样可以预处理出两个环的具体数值。
剩下的就交给FFT吧,将环倍增,所求即∑(x[i+k]-y[i])^2=∑x[i+k]^2 + ∑y[i]^2 - 2*x[i+k]*y[i]的最小值。
前两项可以预处理出来,最后一项同 bzoj2194 ,转化为卷积来求。
注意平方和不是和的平方。
#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 1 << 20
#define pi acos(-1)
using namespace std;
struct data
{
double x , y;
data() {x = y = 0;}
data(double x0 , double y0) {x = x0 , y = y0;}
data operator+(const data a)const {return data(x + a.x , y + a.y);}
data operator-(const data a)const {return data(x - a.x , y - a.y);}
data operator*(const data a)const {return data(x * a.x - y * a.y , x * a.y + y * a.x);}
}a[N] , b[N];
double sx[N] , sy[N];
void fft(data *a , int n , int flag)
{
int i , j , k;
for(i = k = 0 ; i < n ; i ++ )
{
if(i > k) swap(a[i] , a[k]);
for(j = (n >> 1) ; (k ^= j) < j ; j >>= 1);
}
for(k = 2 ; k <= n ; k <<= 1)
{
data wn(cos(2 * pi * flag / k) , sin(2 * pi * flag / k));
for(i = 0 ; i < n ; i += k)
{
data t , w(1 , 0);
for(j = i ; j < i + (k >> 1) ; j ++ , w = w * wn)
t = w * a[j + (k >> 1)] , a[j + (k >> 1)] = a[j] - t , a[j] = a[j] + t;
}
}
}
int main()
{
int n , i , len;
double c = 0 , ans = 10000000000 , sumx = 0 , sumy = 0;
scanf("%d%*d" , &n);
for(i = 0 ; i < n ; i ++ ) scanf("%lf" , &sx[i]) , c -= sx[i];
for(i = 0 ; i < n ; i ++ ) scanf("%lf" , &sy[i]) , c += sy[i];
c = round(c / n);
for(i = 0 ; i < n ; i ++ ) sumx += (sx[i] + c) * (sx[i] + c) , sumy += sy[i] * sy[i];
for(i = 0 ; i < 2 * n ; i ++ ) a[i].x = sx[i % n] + c;
for(i = 0 ; i < n ; i ++ ) b[i].x = sy[n - i - 1];
for(len = 1 ; len < 2 * n ; len <<= 1);
fft(a , len , 1) , fft(b , len , 1);
for(i = 0 ; i < len ; i ++ ) a[i] = a[i] * b[i];
fft(a , len , -1);
for(i = n - 1 ; i < 2 * n - 1 ; i ++ ) ans = min(ans , sumx + sumy - 2 * round(a[i].x / len));
printf("%.0lf\n" , ans);
return 0;
}
原文:http://www.cnblogs.com/GXZlegend/p/6878678.html