洛谷 P3723 [AH/HNOI2017]礼物(bzoj P4827 [Hnoi2017]礼物)

传送门

当初省选时看这题一脸懵逼，打了70分的代码事实满分还是很容易的（啪啪啪，自己wa了十次还说容易？？

好了说下题解吧，这题就是求循环卷积，我的代码是0~n-1，式子是1~n（主要是博主FFT习惯写0~n-1），照样发图片（博主太蠢，不会输数学公式。。）

注意：比较大小时的精度！！！在这里wa了10次！！！

Tips：round()四舍五入函数

代码：

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
using namespace std;
const int Maxn=1000005;
const int oo=2147483647;
struct complex
{
	double real,image;
	complex()
	{
		real=image=0.0;
	}
	complex(double a,double b)
	{
		real=a;
		image=b;
	}
}a[Maxn],b[Maxn],c[Maxn],A[Maxn],x,y;
complex operator +(complex a,complex b)
{
	return complex(a.real+b.real,a.image+b.image);
}
complex operator -(complex a,complex b)
{
	return complex(a.real-b.real,a.image-b.image);
}
complex operator *(complex a,complex b)
{
	return complex(a.real*b.real-a.image*b.image,a.real*b.image+a.image*b.real);
} 
int p[Maxn],q[Maxn],rev[Maxn],dig[Maxn];
int ans1=0,sumx,sumy,N,L,ans2=-oo,ans=oo,ansx,ansy;
int FFT(complex a[],int flag)
{
	for(int i=0;i<N;i++)
		A[i]=a[rev[i]];
	for(int i=0;i<N;i++)
		a[i]=A[i];
	for(int i=2;i<=N;i<<=1)
	{
		complex wn(cos(2*M_PI/i),flag*sin(2*M_PI/i));
		for(int k=0;k<N;k+=i)
		{
			complex w(1,0);
			for(int j=0;j<i/2;j++)
			{
				x=a[k+j];
				y=w*a[k+j+i/2];
				a[k+j]=x+y;
				a[k+j+i/2]=x-y;
				w=w*wn;
			}
		}
	}
	if(flag==-1)
		for(int i=0;i<N;i++)
			a[i].real=a[i].real/N;
}
int main()
{
	freopen("s(2).in","r",stdin);
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
		scanf("%d",&p[i]);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&q[i]);
		q[i+n]=q[i];
	}
	for(int i=0;i<n;i++)
		ans1=ans1+(p[i]*p[i]+q[i]*q[i]);
	for(L=0,N=1;N<2*n;N<<=1,L++);
	N<<=1;
	L++;
	for(int i=0;i<N;i++) 
	{
		int len=0;
		for(int t=i;t;t>>=1)
			dig[len++]=t&1;
		for(int j=0;j<L;j++)
			rev[i]=rev[i]*2+dig[j];
	}
	for(int i=0;i<n;i++)
		a[i]=complex(p[n-i-1],0);
	for(int i=0;i<2*n;i++)
		b[i]=complex(q[i],0);
	FFT(a,1);
	FFT(b,1);
	for(int i=0;i<N;i++)
		c[i]=a[i]*b[i];
	FFT(c,-1);
	for(int i=0;i<n;i++)
	{
		ans2=max(ans2,(int)round(c[n+i-1].real));
		ansx+=p[i];
		ansy+=q[i];
	} 
	for(int i=-m-1;i<=m+1;i++)
	{
		int ans3=ans1-2*ans2;
		ans=min(ans,ans3+2*i*ansx-2*i*ansy+n*i*i);
	}
	printf("%d",ans);
	return 0;
}