洛谷 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)
{

4000
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;
}