题意:
两个数列,每个数列都可以顺序旋转,也可以对所有数同时增加一个非负整数,现在问
的最小值
题解:
在求卷积前要将A数组倍长,B数组翻转
B数组翻转好理解,为什么A数组倍长,因为题目的数列是可以移动的,而我们不知道哪一部分和B卷积是最佳答案,所以讲A数组倍增,然后每次取A中连续长度为n的区间与B倍增,最后取最小值即为最佳答案
因为c是我们设的变量,所以最后循环c找式子最小值
关于FFT求卷积(多项式乘法)
如果有两个无限序列a和b,那么它们卷积的结果是
如果a和b是有限序列,a最低的项为a0,最高的项为an,b同理,我们可以把a和b超出范围的项都设置成0。
套上FFT的模板就可以
代码:
#include <cmath> #include <cctype> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; const int maxn=4e5+5; const double pi=3.1415926535898; int t, n, m, len=1, l, r[maxn*2]; int a[maxn],b[maxn]; struct Cpx{ //复数 double x, y; Cpx (double t1=0, double t2=0){ x=t1, y=t2; } }A[maxn*2], B[maxn*2], C[maxn*2]; Cpx operator +(Cpx a, Cpx b){ return Cpx(a.x+b.x, a.y+b.y); } Cpx operator -(Cpx a, Cpx b){ return Cpx(a.x-b.x, a.y-b.y); } Cpx operator *(Cpx a, Cpx b){ return Cpx(a.x*b.x-a.y*b.y, a.x*b.y+a.y*b.x); } void fdft(Cpx *a, int n, int flag){ //快速将当前多项式从系数表达转换为点值表达 for (int i=0; i<n; ++i) if (i<r[i]) swap(a[i], a[r[i]]); for (int mid=1; mid<n; mid<<=1){ //当前区间长度的一半 Cpx w1(cos(pi/mid), flag*sin(pi/mid)), x, y; for (int j=0; j<n; j+=(mid<<1)){ //j:区间起始点 Cpx w(1, 0); for (int k=0; k<mid; ++k, w=w*w1){ //系数转点值 x=a[j+k], y=w*a[j+mid+k]; a[j+k]=x+y; a[j+mid+k]=x-y; } } } } inline int getint(int &x){ char c; int flag=0; for (c=getchar(); !isdigit(c); c=getchar()) if (c=='-') flag=1; for (x=c-48; c=getchar(), isdigit(c);) x=(x<<3)+(x<<1)+c-48; return flag?x:-x; } int main(){ getint(n); getint(m); int x; int m1=m; m=n; ll A1=0,A2=0,B1=0,B2=0; for (int i=1; i<=n; ++i) { getint(x); A[i].x=x; A[i+n].x=x; A1+=A[i].x*A[i].x; A2+=A[i].x; } // cout<<A1<<" "<<A2<<endl; for (int i=1; i<=m; ++i) { getint(x); B[i].x=x; B1+=B[i].x*B[i].x; B2+=B[i].x; b[i]=B[i].x; } // cout<<B1<<" "<<B2<<endl; for(int i=1;i<=n;i++) { B[i].x=b[n-i+1]; } ll sum=1e15; while (len<=n+n+m) len<<=1, ++l; //idft需要至少l1+l2个点值 for (int i=0; i<len; ++i) //编号的字节长度为l r[i]=(r[i>>1]>>1)|((i&1)<<(l-1)); fdft(A, len, 1); fdft(B, len, 1); for (int i=0; i<=len; ++i) C[i]=A[i]*B[i]; fdft(C, len, -1); //idft for (int i=0; i<=len; ++i)C[i].x=int(C[i].x/len+0.5); for(int i=1;i<=n;i++) { for(int j=-m1;j<=m1;j++) { sum=min(A1+B1+2ll*j*A2-2ll*j*B2+n*j*j-2ll*(ll)C[i+n].x,sum); // cout<<"sum="<<-2ll*(ll)A[i+n].x<<endl; } } cout<<sum; return 0; }