SPOJ SWERC14C Golf Bot

最新推荐文章于 2021-07-16 10:59:47 发布

wang3312362136

最新推荐文章于 2021-07-16 10:59:47 发布

阅读量259

点赞数

CC 4.0 BY-SA版权

分类专栏： FFT

本文链接：https://blog.youkuaiyun.com/wang3312362136/article/details/79547500

FFT 专栏收录该内容

8 篇文章

订阅专栏

本文深入浅出地讲解了FFT算法的基本原理，并通过一个具体的实战案例进行演示。介绍了如何利用FFT算法解决多项式乘法问题，包括算法的核心思想、代码实现细节及优化技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

http://www.spoj.com/problems/SWERC14C/

思路

FFT半裸题了。
如果一个多项式有 $x^i$ 项，另一个多项式有 $x^j$ 项，那么相乘后的多项式就一定有 $x^{i+j}$ 项。
那么如果读入一个数 $i$ ，那么多项式 $x^i$ 项就赋值为。
最后把多项式平方，得出多项式次数为哪些时，系数有值，说明这个次数能被凑出来。

代码

#include <cstdio>
#include <cmath>
#include <algorithm>

const int maxn=200000;
const double pi=acos(-1);

struct complex
{
  double r,i;

  complex(double r_=0,double i_=0)
  {
    r=r_;
    i=i_;
  }

  complex operator +(const complex &other) const
  {
    return complex(r+other.r,i+other.i);
  }

  complex operator -(const complex &other) const
  {
    return complex(r-other.r,i-other.i);
  }

  complex operator *(const complex &other) const
  {
    return complex(r*other.r-i*other.i,r*other.i+i*other.r);
  }
};

complex a[maxn<<2];
int rev[maxn<<2],n,m,ans;

int fft(complex* ar,int len,int op)
{
  for(register int i=0; i<len; ++i)
    {
      if(rev[i]<i)
        {
          std::swap(ar[rev[i]],ar[i]);
        }
    }
  for(register int i=2; i<=len; i<<=1)
    {
      complex wn(cos(2*pi/i),sin(2*pi*op/i));
      for(register int j=0; j<len; j+=i)
        {
          complex w(1,0);
          for(register int k=0; k<(i>>1); ++k)
            {
              complex x=ar[j+k],y=w*ar[j+k+(i>>1)];
              ar[j+k]=x+y;
              ar[j+k+(i>>1)]=x-y;
              w=w*wn;
            }
        }
    }
  if(op==-1)
    {
      for(register int i=0; i<len; ++i)
        {
          ar[i].r/=len;
        }
    }
  return 0;
}

inline int calc(int x)
{
  int l=0;
  m=1;
  while(m<=x)
    {
      m<<=1;
      ++l;
    }
  for(register int i=0; i<m; ++i)
    {
      rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    }
  return 0;
}

int main()
{
  scanf("%d",&n);
  for(register int i=1; i<=n; ++i)
    {
      int w;
      scanf("%d",&w);
      a[w].r=1;
    }
  a[0]=1;
  calc(maxn<<1);
  fft(a,m,1);
  for(register int i=0; i<m; ++i)
    {
      a[i]=a[i]*a[i];
    }
  fft(a,m,-1);
  scanf("%d",&n);
  for(register int i=1; i<=n; ++i)
    {
      int w;
      scanf("%d",&w);
      if(a[w].r-0.5>0)
        {
          ++ans;
        }
    }
  printf("%d\n",ans);
  return 0;
}