题目大意:计算 c[k]=∑k≤i<na[i]×b[i−k]
题解:令
a[n−i−1]=a[i],i∈[0,n)
c[k]=∑k≤i<na[n−i−1]×b[i−k]
令 d[n−k−1]=∑k≤i<na[n−i−1]×b[i−k]
然后就可以卷积了……
最后 c[0…n−1] 对应的是 d[n−1…0]
我的收获:构造卷积……
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <complex>
using namespace std;
#define pi acos(-1)
typedef complex<double> C;
const int M=262145;
int n,m,L,R[M];
C a[M],b[M];
void pre()
{
m=n<<1;//相乘后的位数是原来的2倍
for(n=1;n<=m;n<<=1) L++;
for(int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));//求逆序表:末位为0,直接为其前一半逆序表的值右移一位,末位为1,在最高位添加1
}
void FFT(C *a,int f=1){
for(int i=0;i<n;i++) if(R[i]>i) swap(a[i],a[R[i]]);//利用逆序表,快速求逆序
for(int i=1;i<n;i<<=1){
C wn(cos(pi/i),f*sin(pi/i));
for(int j=0;j<n;j+=i<<1){
C w(1,0);for(int k=0;k<i;k++,w*=wn){
C x=a[j+k],y=w*a[j+k+i];
a[j+k]=x+y;a[j+k+i]=x-y;
}
}
}
if(f==-1) for(int i=0;i<n;i++) a[i]/=n;
}
void print(){
for(int i=m/2-1;i>=0;i--) printf("%d\n",(int)(a[i].real()+0.5));
}
void work()
{
FFT(a,1),FFT(b,1);
for(int i=0;i<n;i++) a[i]*=b[i];
FFT(a,-1);
print();
}
void init()
{
scanf("%d",&n);
for(int x,i=0;i<n;i++){
scanf("%d",&x);a[n-i-1]=x;
scanf("%d",&x);b[i]=x;
}
pre();
}
int main()
{
init();
work();
return 0;
}