luoguP4173 残缺的字符串 FFT

链接

luogu

思路

和昨天做的题几乎一样。
匹配等价于（其实我更喜欢fft从0开始）
\(\sum\limits_{i=0}^{m-1}(S[i+j]-T[i])^2*T[i]*S[i]=0\)
展开得到
\(\sum\limits_{i=0}^{m-1}S_{i+j}^3T_{i}+\sum\limits\S_{i+j}T_{i}^3-2\sum\limits\S_{i+j}^2T_{i}^2=0\)
反转T串串
\(\sum\limits_{i=0}^{m-1}S_{i+j}^3T_{m-1-i}+\sum\limits\S_{i+j}T_{m-1-i}^3-2\sum\limits\S_{i+j}^2T_{m-1-i}^2=0\)
然后三次卷积就行了。
写完看了看别的代码，感觉自己的代码太蠢了，多fft了好几次。

错误

我居然同样的错误犯了好几遍。
昨天也是wrong在最后从0到n-m判断，这次又是。
而且我还想了好久。

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1200000+7;
const double Pi=acos(-1),eps=1e-9;
int limit=1,p,r[N<<1];
int fff(int x) {return x*x*x;}
int ff(int x) {return x*x;}
struct Complex {double x,y;Complex(double xx=0,double yy=0) {x=xx,y=yy;}
}a[N<<1],b[N<<1],tot[N<<1];
Complex operator + (Complex a,Complex b) {return Complex(a.x+b.x,a.y+b.y);}
Complex operator - (Complex a,Complex b) {return Complex(a.x-b.x,a.y-b.y);}
Complex operator * (Complex a,Complex b) {return Complex(a.x*b.x-a.y*b.y,a.y*b.x+b.y*a.x);}
void fft(Complex *a,int type) {for(int i=0;i<limit;++i)if(i<r[i]) swap(a[r[i]],a[i]);for(int mid=1;mid<limit;mid<<=1) {Complex Wn(cos(Pi/mid),type*sin(Pi/mid));for(int i=0;i<limit;i+=mid<<1) {Complex w(1,0);for(int j=0;j<mid;++j,w=w*Wn) {Complex x=a[i+j],y=w*a[i+j+mid];a[i+j]=x+y;a[i+j+mid]=x-y;}}}if(type==-1) for(int i=0;i<limit;++i) a[i].x/=limit;
}
int n,m,s[N],t[N];
char S[N],T[N];
int main() {scanf("%d%d%s%s",&m,&n,T,S);for(int i=0;i<n;++i) s[i]=S[i]=='*'?0:S[i]-'a'+1;for(int i=0;i<m;++i) t[m-i-1]=T[i]=='*'?0:T[i]-'a'+1;while(limit<n+m) limit<<=1,p++;for(int i=0;i<limit;++i)r[i]=(r[i>>1]>>1)|((i&1)<<(p-1));for(int i=0;i<limit;++i) a[i]=Complex(s[i],0),b[i]=Complex(fff(t[i]),0);fft(a,1),fft(b,1);for(int i=0;i<limit;++i) tot[i]=tot[i]+a[i]*b[i];for(int i=0;i<limit;++i) a[i]=Complex(fff(s[i]),0),b[i]=Complex(t[i],0);fft(a,1),fft(b,1);for(int i=0;i<limit;++i) tot[i]=tot[i]+a[i]*b[i];for(int i=0;i<limit;++i) a[i]=Complex(ff(s[i]),0),b[i]=Complex(ff(t[i]),0);fft(a,1),fft(b,1);for(int i=0;i<limit;++i) tot[i]=tot[i]-a[i]*b[i]*Complex(2,0);fft(tot,-1);int ans=0;for(int i=0;i<n-m+1;++i) if(fabs(round(tot[m+i-1].x))<eps) ans++;printf("%d\n",ans);for(int i=0;i<n-m+1;++i) if(fabs(round(tot[m+i-1].x))<eps) printf("%d ",i+1);return 0;
}