[BZOJ4259][FFT]残缺的字符串

复习一下FFT
http://blog.csdn.net/u011542204/article/details/50708834
感觉这种字符串匹配的方法很棒啊
然而我的代码被卡常卡内存……

#include 
#include 
#include 
#include 
#define eps 1e-7
#define N 1100010

using namespace std;

const double pi=acos(-1);

struct E{
  double real,imag;
  E(double r=0,double i=0):real(r),imag(i){}
  friend E operator +(const E &a,const E &b){ return E(a.real+b.real,a.imag+b.imag); }
  friend E operator -(const E &a,const E &b){ return E(a.real-b.real,a.imag-b.imag); }
  friend E operator *(const E &a,const E &b){
    return E(a.real*b.real-a.imag*b.imag,a.real*b.imag+a.imag*b.real);
  }
  friend E operator /(const E &a,const double &b){ return E(a.real/b,a.imag/b); }
};

int n,m,L,M,p;
int rev[N],Ans[N];
char A[N],B[N];
E a[N],b[N],a1[N],b1[N],f[N];

inline void FFT(E *a,int r){
  for(int i=0;iif(rev[i]>i) swap(a[i],a[rev[i]]);
  for(int i=1;i1){
    E wn(cos(pi/i),r*sin(pi/i));
    for(int j=0;j1)){
      E w(1,0);
      for(int k=0;kif(r==-1) for(int i=0;iint main(){
  scanf("%d %d",&n,&m); int in=n;
  scanf("%s%s",A,B);
  reverse(A,A+n);
  for(int i=0;i'*')?0:(A[i]-'a'+1);
  for(int i=0;i'*')?0:(B[i]-'a'+1);
  for(n=1,M=m*2;n<=M;n<<=1,L++);
  for(int i=0;i>1]>>1)|((i&1)<1);
  for(int i=0;i1); FFT(b1,1);
  for(int i=0;ifor(int i=0;i1); FFT(b1,1);
  for(int i=0;ifor(int i=0;i1); FFT(b1,1);
  for(int i=0;i1);
  for(int i=in-1;iif(int(0.1+f[i].real)==0) Ans[++p]=i;
  printf("%d\n",p);
  for(int i=1;i<=p;i++) printf("%d%c",Ans[i]-in+2,i==p?'\n':' ');
  return 0;
}

你可能感兴趣的:(FFT,&,FWT)