P4173 残缺的字符串
题目链接
题意分析
啥 ? ? ? \(FFT\)做字符串匹配
可是就是这样
我们定义匹配函数
我们定义\(A\)是匹配串 \(B\)是被匹配串
我们当前到达\(B\)串的\(x\)位置
\[P(x)=\sum_{i=0}^{m-1}[A(i)-B(x-m+i+1)]
\]
那么如果\(P(x)=0\)的话 是不是就是匹配上了呢? ?
不是 如果匹配的是\(ab\)以及\(ba\)的话 那么就\(WA\)了
所以我们令
\[P(x)=\sum_{i=0}^{m-1}[A(i)-B(x-m+i+1)]^2
\]
\[=\sum_{i=0}^{m-1}A^2(i)+\sum_{i=0}^{m-1}B^2(x-m+i+1)-2*\sum_{i=0}^{m-1}A(i)B(x-m+i+1)
\]
我们如果颠倒一下就是 \(S(m-i-1)=S(i)\)
\[P(x)=\sum_{i=0}^{m-1}S^2(m-i+1)+\sum_{i=0}^{m-1}B^2(x-m+i+1)-2*\sum_{i=0}^{m-1}S(m-i-1)B(x-m+i+1)
\]
首先第一项我们可以作为常量处理
第二项我们可以使用前缀和处理
第三项就是
\[\sum_{i+j=x}S(i)B(j)
\]
我们可以使用\(FFT\)来做
如果加上了通配符呢 ? ? ?
我们就改良匹配公式
\[P(x)=\sum_{i=0}^{m-1}[A(i)-B(x-m+i+1)]^2A(i)B(x-m+i+1)
\]
然后就是暴力展开
\[P(x)=\sum_{i=0}^{m-1}S^3(m-i-1)B(x-m+i+1)+\sum_{i=0}^{m-1}S(m-i-1)B^3(x-m+i+1)-2*\sum_{i=0}^{m-1}S^2(m-i-1)B(x-m+i+1)
\]
我们弄成卷积的形式
\[P(x)=\sum_{i+j=x}S^3(i)B(j)+\sum_{i+j=x}S(i)B^3(j)-2*\sum_{i+j=x}S^2(i)B^2(j)
\]
然后还是跑\(FFT\)
CODE:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<string>
#include<queue>
#include<map>
#include<stack>
#include<list>
#include<set>
#include<deque>
#include<vector>
#include<ctime>
#define ll long long
#define inf 0x7fffffff
#define N 500008
#define IL inline
#define M 2008611
#define D double
#define eps 0.5
#define R register
using namespace std;
template<typename T>IL void read(T &_)
{
T __=0,___=1;char ____=getchar();
while(!isdigit(____)) {if(____=='-') ___=0;____=getchar();}
while(isdigit(____)) {__=(__<<1)+(__<<3)+____-'0';____=getchar();}
_=___ ? __:-__;
}
/*-------------OI使我快乐-------------*/
const D Pi=acos(-1.0);
int n,m,lim,all,top;D tot;
char s1[M],s2[M];
D sx[M],sy[M];int key[M],sta[M];
struct Node{
D xx,yy;
Node(D xxx=0,D yyy=0){xx=xxx;yy=yyy;}
friend Node operator +(const Node &A,const Node &B)
{return (Node){A.xx+B.xx,A.yy+B.yy};}
friend Node operator -(const Node &A,const Node &B)
{return (Node){A.xx-B.xx,A.yy-B.yy};}
friend Node operator *(const Node &A,const Node &B)
{return (Node){A.xx*B.xx-A.yy*B.yy,A.xx*B.yy+A.yy*B.xx};}
friend Node operator /(const Node &A,const D &B)
{return (Node){A.xx/B,A.yy/B};}
friend Node operator *(const Node &A,const D &B)
{return (Node){A.xx*B,A.yy*B};}
}cdy[M],wzy[M],ans[M];
IL void FFT(Node *A,int knd)
{
for(R int i=0;i<lim;++i) if(i<key[i]) swap(A[i],A[key[i]]);
for(R int mid=1;mid<lim;mid<<=1)
{
Node Wn=(Node){cos(Pi/mid),knd*sin(Pi/mid)};
for(R int j=0;j<lim;j+=(mid<<1))
{
Node w=(Node){1,0};
for(R int k=0;k<mid;++k,w=w*Wn)
{
Node x=A[j+k],y=w*A[j+k+mid];
A[j+k]=x+y;A[j+k+mid]=x-y;
}
}
}
if(knd==-1)
for(R int i=0;i<lim;++i)
A[i]=A[i]/lim;
}
int main()
{
// freopen(".in","r",stdin);
// freopen(".out","w",stdout);
read(n);read(m);scanf("%s%s",s1,s2);
for(lim=1;lim<(m<<1);lim<<=1) ++all;
for(R int i=0;i<lim;++i) key[i]=((key[i>>1]>>1)|((i&1)<<(all-1)));
for(R int i=0;i<n;++i) if(s1[i]!='*') sx[i]=(s1[i]-'a'+1);
for(R int i=0;i<m;++i) if(s2[i]!='*') sy[i]=(s2[i]-'a'+1);
reverse(sx,sx+n);
for(R int i=0;i<n;++i) cdy[i]=(Node){sx[i]*sx[i]*sx[i],0};
for(R int i=0;i<m;++i) wzy[i]=(Node){sy[i],0};
FFT(cdy,1);FFT(wzy,1);
for(R int i=0;i<lim;++i) ans[i]=ans[i]+cdy[i]*wzy[i];
// for(R int i=0;i<lim;++i) printf("%.4f\n",ans[i].xx);
for(R int i=0;i<lim;++i) cdy[i]=wzy[i]=(Node){0,0};
for(R int i=0;i<n;++i) cdy[i]=(Node){sx[i]*sx[i],0};
for(R int i=0;i<m;++i) wzy[i]=(Node){sy[i]*sy[i],0};
FFT(cdy,1);FFT(wzy,1);
for(R int i=0;i<lim;++i) ans[i]=ans[i]-cdy[i]*wzy[i]*2.0;
// for(R int i=0;i<lim;++i) printf("%.4f\n",ans[i].xx);
for(R int i=0;i<lim;++i) cdy[i]=wzy[i]=(Node){0,0};
for(R int i=0;i<n;++i) cdy[i]=(Node){sx[i],0};
for(R int i=0;i<m;++i) wzy[i]=(Node){sy[i]*sy[i]*sy[i],0};
FFT(cdy,1);FFT(wzy,1);
for(R int i=0;i<lim;++i) ans[i]=ans[i]+cdy[i]*wzy[i];
// for(R int i=0;i<lim;++i) printf("%.4f\n",ans[i].xx);
FFT(ans,-1);
// for(R int i=0;i<lim;++i) printf("%.4f\n",ans[i].xx);
for(R int i=n-1;i<m;++i)
if(fabs(ans[i].xx)<eps) sta[++top]=i-n+2;
printf("%d\n",top);
for(R int i=1;i<=top;++i)
printf("%d%c",sta[i],(i==top ? '\n':' '));
// fclose(stdin);
// fclose(stdout);
return 0;
}
