「AHOI / HNOI2017」礼物

「AHOI / HNOI2017」礼物

题目描述

我的室友最近喜欢上了一个可爱的小女生。马上就要到她的生日了，他决定买一对情侣手环，一个留给自己，一个送给她。每个手环上各有 n 个装饰物，并且每个装饰物都有一定的亮度。

但是在她生日的前一天，我的室友突然发现他好像拿错了一个手环，而且已经没时间去更换它了！他只能使用一种特殊的方法，将其中一个手环中所有装饰物的亮度增加一个相同的自然数 c（即非负整数）。并且由于这个手环是一个圆，可以以任意的角度旋转它，但是由于上面装饰物的方向是固定的，所以手环不能翻转。需要在经过亮度改造和旋转之后，使得两个手环的差异值最小。

在将两个手环旋转且装饰物对齐了之后，从对齐的某个位置开始逆时针方向对装饰物编号1,2,…,n，其中 n 为每个手环的装饰物个数， 第 1 个手环的 i 号位置装饰物亮度为 xi，第 2 个手环的 i 号位置装饰物亮度为 yi，两个手环之间的差异值为(参见输入输出样例和样例解释)：

$\sum_{i=1}^{n} (x_i-y_i)^2$

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

输入输出格式

输入格式：

输入数据的第一行有两个数n, m，代表每条手环的装饰物的数量为n，每个装饰物的初始亮度小于等于m。

接下来两行，每行各有n个数，分别代表第一条手环和第二条手环上从某个位置开始逆时针方向上各装饰物的亮度。

输出格式：

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

可以大于 m。

输入输出样例

输入样例#1：
复制

5 6
1 2 3 4 5
6 3 3 4 5

输出样例#1：
复制

1

说明

【样例解释】

需要将第一个手环的亮度增加1，第一个手环的亮度变为： 2 3 4 5 6

旋转一下第二个手环。对于该样例，是将第二个手环的亮度6 3 3 4 5向左循环移动一个位置，使得第二手环的最终的亮度为： 3 3 4 5 6。

此时两个手环的亮度差异值为1

【数据范围】

30%的数据满足n≤500, m≤10；

70%的数据满足n≤5000；

100%的数据满足1≤n≤50000, 1≤m≤100, 1≤ai≤m。

Soda的题解

首先，有一个结论：两个手环增加非负整数亮度，等于其中一个增加一个整数亮度（可以为负）。也就是相对性。

我们令增加量为\(x\)，旋转以后的原数列为\({a}{b}\)那么现在的费用就是：

\[\sum_{i=1}^n\left(a_i+x-b_i\right)^2
\]

我们把第\(i\)项拿出来拆开，得到：

\[\left(a_i+x-b_i\right)^2=a_i^2+b_i^2+x^2+2a_ix-2a_ib_i-2b_ix
\]

那么原式变成了

\[\sum_{i=1}^na_i^2+\sum_{i=1}^nb_i^2+nx^2+2x\left(\sum_{i=1}^na_i-\sum_{i=1}^nb_i\right)-2\sum_{i=1}^na_ib_i
\]

我们发现，这个式子除了最后一项之外都是确定的。那么我们只要令最后一项最大，那么就可以得到最小的费用值了

现在问题转化为求\(\sum_{i=1}^na_ib_i\)的最大值

等等，这个形式......

我们把数列{a}反过来，变成

\[\sum_{i=1}^na_{n-i+1}b_i
\]

这是一个卷积吗

所以把反过来的数列\({a}\)倍长，和数列\({b}\)卷积，得到的项里面的第\(n+1\)到\(n*2\)项的最大值，就是\(\sum_{i=1}^na_ib_i\)的最大值

然后把前面的不变项加上，枚举变化量，就是答案了

时间复杂度\(O(n\log n+nm)\)

struct node {double x,y;};
il node operator+(co node&a,co node&b){
	return (node){a.x+b.x,a.y+b.y};
}
il node operator-(co node&a,co node&b){
	return (node){a.x-b.x,a.y-b.y};
}
il node operator*(co node&a,co node&b){
	return (node){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};
}
il node operator/(co node&a,double k){
	return (node){a.x/k,a.y/k};
}
co double PI=acos(-1);
co int N=262145; // 2^18+1
int rev[N];
void fourier_trans(node a[],int limit,int inverse){
	for(int i=0;i<limit;++i)if(i<rev[i]) swap(a[i],a[rev[i]]);
	for(int step=1;step<limit;step<<=1){
		double alpha=inverse*PI/step;
		for(int k=0;k<step;++k){
			node omega=(node){cos(alpha*k),sin(alpha*k)};
			for(int even=k;even<limit;even+=step<<1){
				int odd=even+step;
				node t=omega*a[odd];
				a[odd]=a[even]-t,a[even]=a[even]+t;
			}
		}
	}
	if(inverse==-1)for(int i=0;i<limit;++i) a[i]=a[i]/limit;
}
int n,m;
node a[N],b[N];
LL s1,s2,ans=1e18;
int main(){
	read(n),read(m);
	for(int i=1;i<=n;++i){
		read(a[i].x);
		s1+=a[i].x*a[i].x,s2+=a[i].x;
	}
	for(int i=1;i<=n;++i){
		read(b[i].x);
		s1+=b[i].x*b[i].x,s2-=b[i].x;
	}
	copy(a+1,a+n+1,a+n+1);
	reverse(b+1,b+n+1);
	int len=ceil(log2(3*n+1)),limit=1<<len;
	for(int i=0;i<limit;++i) rev[i]=rev[i>>1]>>1|(i&1)<<(len-1);
	fourier_trans(a,limit,1),fourier_trans(b,limit,1);
	for(int i=0;i<limit;++i) a[i]=a[i]*b[i];
	fourier_trans(a,limit,-1);
	for(int i=0;i<limit;++i) a[i].x=round(a[i].x);
	for(int i=1;i<=n;++i)
		for(int j=-m;j<=m;++j)
			ans=min(ans,s1+j*j*n+2*j*s2-2*(LL)a[i+n].x);
	printf("%lld\n",ans);
	return 0;
}

体验了一下FFT用手写complex和数组的写法，久违了。