bzoj4827: [Hnoi2017]礼物

看到要求∑(xi − y i+k -c )2的最小值。首先将xi − y i+k看作整体，得到一个关于c的二次函数nc^2+pc+q=0。发现q的形式将y数组翻转类似一个卷积，然后又发现其中每个值小于998244353，然后一个NTT上去就能求出精确的最小值了。O(nlogn)

#include<iostream>
#include<cstdio>
#define P 998244353
#define N 200005
#define ll long long
using namespace std;
int n,m,a[N],b[N],A1,A2,w[2][N],r[N];
int pow(int x,int y,int p)
{
	int Ans=1;
	for (;y;x=(ll)x*x%p,y>>=1)
		if (y&1) Ans=(ll)Ans*x%p;
	return Ans;
}
int tl,tn,rn;
void pre(int n)
{
	tl=0,tn=1;
	for (;tn<n;tl++,tn<<=1);
	tl++;tn<<=1;
	int W=pow(3,(P-1)/tn,P);
	w[0][0]=w[1][0]=1;
	for (int i=1;i<tn;i++)
		w[0][i]=(ll)w[0][i-1]*W%P;
	for (int i=1;i<tn;i++)
		w[1][i]=w[0][tn-i];
	for (int i=0;i<tn;i++)
		r[i]=(r[i>>1]>>1)|((i&1)<<(tl-1));
	rn=pow(tn,P-2,P);
}
void dft(int a[],int p)
{
	for (int i=0;i<tn;i++)
		if (r[i]>i) swap(a[i],a[r[i]]);
	
	for (int i=1;i<tn;i<<=1)
	for (int j=0,t=tn/(i<<1);j<tn;j+=i<<1)
	for (int k=0,l=0;k<i;k++,l+=t)
	{
		int x=(ll)w[p][l]*a[j+k+i]%P;
		int y=a[j+k];
		a[j+k]=(y+x)%P;
		a[j+k+i]=(y+P-x)%P;
	}
		scanf("%d",&a[i]),A1+=a[i],A2=A2+a[i]*a[i];
	for (int i=0;i<n;i++)
	if (p)
		for (int i=0;i<tn;i++)
			a[i]=(ll)a[i]*rn%P;
}
int main()
{
	scanf("%d%d",&n,&m);
	
	for (int i=0;i<n;i++)
		scanf("%d",&b[n-i]),A1-=b[n-i],A2=A2+b[n-i]*b[n-i];
	pre(n);
	dft(a,0);dft(b,0);
	
	for (int i=0;i<tn;i++)
		a[i]=(ll)a[i]*b[i]%P;
	dft(a,1);
	ll Ans=0;
	for (int i=0;i<n;i++)
		Ans=max(Ans,(ll)a[i]+(ll)a[i+n]);

	if (A1<0) A1=-A1;A1*=2;
	int c=A1/(2*n);
	A1=min(n*c*c-A1*c,n*(c+1)*(c+1)-A1*(c+1));
	Ans=-2*Ans+A2+A1;
	printf("%lld\n",Ans);
}