bzoj 3527 [Zjoi2014]力——FFT

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3527

把 q[ i ] 除掉。设 g[ i ] = i^2 ，有一半的式子就变成卷积了；另一半只要翻转一下序列就也变成卷积了。

g[ i ] 那个部分FFT过一次之后就不用再FFT了。

注意别在主函数里把全局变量的 len 覆盖了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define db double
#define ll long long
using namespace std;
const int N=1e5+,M=N<<; const db pi=acos(-);
int n,len,r[M];
db f[N],g[N],ans[N];
struct cpl{db x,y;}a[M],b[M],I;
cpl operator+ (cpl a,cpl b){return (cpl){a.x+b.x,a.y+b.y};}
cpl operator- (cpl a,cpl b){return (cpl){a.x-b.x,a.y-b.y};}
cpl operator* (cpl a,cpl b){return (cpl){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};}
void fft(cpl *a,bool fx)
{
  for(int i=;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=;R<=len;R<<=)
    {
      int m=R>>;
      cpl Wn=(cpl){ cos(pi/m),fx?-sin(pi/m):sin(pi/m) };
      for(int i=;i<len;i+=R)
    {
      cpl w=I;
      for(int j=;j<m;j++,w=w*Wn)
        {
          cpl tmp=w*a[i+m+j];
          a[i+m+j]=a[i+j]-tmp;
          a[i+j]=a[i+j]+tmp;
        }
    }
    }
}
int main()
{
  scanf("%d",&n); I.x=;
  for(int i=;i<n;i++)scanf("%lf",&f[i]);
  for(int i=;i<n;i++)g[i]=(db)/i/i;
  for(int i=;i<n;i++)
    a[i].x=f[i],b[i].x=g[i];
  len=;//do not 'int len'!!!!!
  for(;len<=n<<;len<<=);
  for(int i=;i<len;i++)r[i]=(r[i>>]>>)+((i&)?len>>:);
  fft(a,); fft(b,);
  for(int i=;i<len;i++)a[i]=a[i]*b[i];
  fft(a,);
  for(int i=;i<n;i++) ans[i]=a[i].x/len;////// /len!!!

  for(int i=;i<len;i++) a[i].x=a[i].y=;
  for(int i=;i<n;i++) a[i].x=f[n--i];
  fft(a,);
  for(int i=;i<len;i++) a[i]=a[i]*b[i];
  fft(a,);
  for(int i=;i<n;i++)
    {
      ans[i]-=a[n--i].x/len;
      printf("%.3f\n",ans[i]);
    }
  return ;
}