BZOJ.4827.[HNOI2017]礼物(FFT)

BZOJ
洛谷


首先不妨枚举旋转多少次(k),然后题目就是要求(sum_{i=1}^{n}(x_{i+k}-y_i+c)^2)的最小值,其中(kin[0,n))(cin[-m,m])(显然(c)枚举到(m)就足够了,而另一个环增加可以看成这个环减少),且(x_{i+n}=x_i)(是个环啊)。
把式子拆开:$$sum_{i=1}^{n}(x_{i+k}-y_i+c)^2=sum_{i=1}^{n}left[x_{i+k}^2+y_i^2-2x_{i+k}y_i+c^2+2c(x_{i+k}-y_i) ight]$$

再把(sum)拆了:$$sum_{i=1}^nx_{i+k}^2+sum_{i=1}^n y_i^2-2sum_{i=1}^n x_{i+k}y_i+nc^2+2cleft(sum_{i=1}^n x_i-sum_{i=1}^n y_i ight)$$

我们发现前面两项(sum_{i=1}^nx_{i+k}^2+sum_{i=1}^n y_i^2)是不变的,后面两项(nc^2+2cleft(sum_{i=1}^n x_i-sum_{i=1}^n y_i ight))在序列已经确定的情况下只与(c)有关,枚举(c)求个最小值就可以了。

然后我们需要求(sum_{i=1}^n x_{i+k}y_i)的最大值,它只与(k)有关。
这不就和某些FFT的题一样么,反转(x)(y)数组,变为求(sum_{i=1}^n x_{i+k}y_{n-i+1}),就是卷积了。(Ans_k)就是多项式相乘后的第(n+k+1)项。
枚举(k)取一个(max{Ans_k})就可以了。


//12380kb	1120ms(为啥我的FFT这么慢啊QAQ)
#include <cmath>
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 300000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=(1<<18)+5;//3n!
const double PI=acos(-1);

int X[N],Y[N],rev[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Complex
{
	double x,y;
	Complex(double x=0,double y=0):x(x),y(y) {}
	Complex operator +(const Complex &a) {return Complex(x+a.x, y+a.y);}
	Complex operator -(const Complex &a) {return Complex(x-a.x, y-a.y);}
	Complex operator *(const Complex &a) {return Complex(x*a.x-y*a.y, x*a.y+y*a.x);}
}A[N],B[N];

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
void FFT(Complex *a,int lim,int opt)
{
	for(int i=1; i<lim; ++i) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
	for(int i=2; i<=lim; i<<=1)
	{
		int mid=i>>1; Complex Wn(cos(PI/mid),opt*sin(PI/mid));
		for(int j=0; j<lim; j+=i)
		{
			Complex w(1,0),t;
			for(int k=j; k<j+mid; ++k,w=w*Wn)
				a[k+mid]=a[k]-(t=w*a[k+mid]), a[k]=a[k]+t;
		}
//		for(int j=0; j<mid; ++j) W[j]=w, w=w*Wn;//预处理果然并不会快...
//		for(int j=0; j<lim; j+=i)
//			for(int k=0; k<mid; ++k)
//				a[j+k+mid]=a[j+k]-(t=W[k]*a[j+k+mid]), a[j+k]=a[j+k]+t;
	}
	if(opt==-1) for(int i=0; i<lim; ++i) a[i].x/=lim;
}
int Calc(int n)
{
	for(int i=1; i<=n; ++i) A[i+n]=A[i]=Complex(X[i],0);
	for(int i=1; i<=n; ++i) B[i]=Complex(Y[n-i+1],0);
	int lim=1,l=-1;
	while(lim<=3*n) lim<<=1,++l;
	for(int i=1; i<lim; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l);
	FFT(A,lim,1), FFT(B,lim,1);
	for(int i=0; i<lim; ++i) A[i]=A[i]*B[i];
	FFT(A,lim,-1);
	int ans=0;
	for(int k=0; k<n; ++k) ans=std::max(ans,(int)(A[n+k+1].x+0.5));
	return ans;
}

int main()
{
	int n=read(),m=read();
	for(int i=1; i<=n; ++i) X[i]=read();
	for(int i=1; i<=n; ++i) Y[i]=read();
	int ans=0,s=0;
	for(int i=1; i<=n; ++i) ans+=X[i]*X[i]+Y[i]*Y[i], s+=X[i]-Y[i];
	int mn=2e9;
	for(int c=-m; c<=m; ++c) mn=std::min(mn,n*c*c+2*c*s);
	ans+=mn-2*Calc(n);
	printf("%d
",ans);

	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/10037221.html