【洛谷4173】残缺的字符串(重拾FFT)

点此看题面

大致题意: 有一个长度为\(n\)的字符串\(A\)和一个长度为\(m\)的字符串\(B\),其中存在一些字符'*'可以与任意字符匹配。求\(B\)中所有满足条件的位置,使得从这一位置开始的连续\(n\)个字符组成的字符串能与\(A\)完全匹配。

前言

之前一直对\(FFT\)这个算法不太理解,最近感觉数学水平有了点小提升,于是去重新看了遍\(FFT\)的原理,突然发现能大致理解了......

这道题原题是\(A\)长度为\(m\)\(B\)长度为\(n\),出于个人习惯,把\(n\)\(m\)调换了一下,变成\(A\)长度为\(n\)\(B\)长度为\(m\)

同样出于个人习惯,我还把\(n\)\(m\)各自减了\(1\),则\(A\)中下标为\(0\sim n\)\(B\)中下标为\(0\sim m\),由于题目要求下标从\(1\)开始,所以最后又将答案加了\(1\)

希望能理解。

\(FFT\)与字符串匹配

\(FFT\)与字符串匹配,看似没有任何联系,实际上却大有用处。

例如这题,首先我们可以把字符转化成数字,'a'~'z'各自对应\(1\sim 26\),而'*'就用\(0\)来表示。

先不考虑'*',此时如果要比较两个字符\(x,y\)是否匹配,显然我们可以发现若\(x=y\),即\(x-y=0\),两个字符匹配,而若\(x-y≠0\),则两个字符不匹配。

也就是说,我们这里把\(x-y\)看作是匹配值,匹配值为\(0\)表示匹配,不为\(0\)表示不匹配。

但是,如果要比较两个字符串是否匹配,却不能单纯把每一位上\(x-y\)的值加一起来判断是否\(0\),因为\(x-y\)可正可负,加一起可能会相互抵消成\(0\)

怎么办?

把负号去掉不就行了吗?首先便会想到把匹配值改为\(|x-y|\),但这似乎不太好操作。

但是,如果改为\((x-y)^2\),似乎就很可行的样子。

总结一下,要比较两个长度相等的字符串是否完全匹配,我们可以对于每一位求出\((x-y)^2\),然后求和,如果最后的和为\(0\),说明完全匹配,否则说明不匹配。

那出现了'*'呢?

由于'*'可以与任意字符匹配,且上面我们决定了用\(0\)来表示''*',而完全匹配的定义又是匹配值为\(0\)

所以,我们可以把匹配值改为\(xy(x-y)^2\),如果这个式子求和为\(0\),就说明完全匹配。

推式子

假设我们现在要判断从\(B\)中第\(i\)位开始的连续\(n\)个字符是否能与\(A\)完全匹配。

则就是要求:

\[\sum_{k=0}^nA_kB_{i+k}(A_k-B_{i+k})^2 \]

拆开得到:

\[\sum_{k=0}^n(A_k^3B_{i+k}-2A_k^2B_{i+k}^2+A_kB_{i+k}^3) \]

考虑设\(T_{k}=A_{n-k}\),即\(A\)串的倒串,则原式等同于:

\[\sum_{k=0}^n(T_{n-k}^3B_{i+k}-2T_{n-k}^2B_{i+k}^2+T_{n-k}B_{i+k}^3) \]

然后我们就能发现一件很神奇的是:每对相乘的项下标之和都是\(n+i\),在\(i\)一定时是个定值!

所以,我们可以把上面的式子改成卷积形式,即:

\[\sum_{x+y=n+i}(T_{x}^3B_{y}-2T_{x}^2B_{y}^2+T_{x}B_{y}^3) \]

这么一来,我们只要对\(T,T^2,T^3,B,B^2,B^3\)六个多项式分别做\(DFT\),进行上述运算后,再做\(IDFT\),即可得到一个新的多项式\(P\)

而从\(B\)中第\(i\)位开始的连续\(n\)个字符能与\(A\)完全匹配,当且仅当\(P\)中第\(n+i\)项的系数为\(0\)

这样这题就做完了。

代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 300000
#define LL long long
#define DB double
using namespace std;
int n,m,a[4*N+5],b[4*N+5];char s[N+5];
template<int SZ> class Poly
{
	private:
		int P,L,R[4*N+5];DB Pi;
		struct node
		{
			DB x,y;I node(Con DB& a=0,Con DB& b=0):x(a),y(b){}
			I node operator + (Con node& o) Con {return node(x+o.x,y+o.y);}
			I node operator - (Con node& o) Con {return node(x-o.x,y-o.y);}
			I node operator * (Con node& o) Con {return node(x*o.x-y*o.y,x*o.y+y*o.x);}
		}A[4*N+5],A2[4*N+5],A3[4*N+5],B[4*N+5],B2[4*N+5],B3[4*N+5];
		I void T(node *s,CI op)
		{
			RI i,j,k;node U,S,x,y;for(i=0;i^P;++i) i<R[i]&&(U=s[i],s[i]=s[R[i]],s[R[i]]=U,0);
			for(i=1;i^P;i<<=1) for(U=node(cos(Pi/i),op*sin(Pi/i)),j=0;j^P;j+=i<<1)
				for(S=node(1,0),k=0;k^i;++k,S=S*U) s[j+k]=(x=s[j+k])+(y=S*s[i+j+k]),s[i+j+k]=x-y;
		}
	public:
		I Poly() {Pi=acos(-1);}
		Tp I void FFT(CI n,CI m,Ty *a,Ty *b)
		{
			RI i;P=1,L=0;W(P<=n+m) P<<=1,++L;for(i=0;i^P;++i) R[i]=(R[i>>1]>>1)|((i&1)<<L-1);
			for(i=0;i<=n;++i) A[i]=a[i],A2[i]=A[i]*A[i],A3[i]=A[i]*A2[i];//记录多项式
			for(i=0;i<=m;++i) B[i]=b[i],B2[i]=B[i]*B[i],B3[i]=B[i]*B2[i];//记录多项式
			T(A,1),T(A2,1),T(A3,1),T(B,1),T(B2,1),T(B3,1);//DFT
			for(i=0;i^P;++i) A[i]=A3[i]*B[i]-node(2)*A2[i]*B2[i]+A[i]*B3[i];//运算
			for(T(A,-1),i=0;i<=m;++i) a[i]=A[i].x/P+0.5;//IDFT
		}
};Poly<N> P;
int main()

	RI i,t=0;scanf("%d%d",&n,&m),--n,--m;
	for(scanf("%s",s),i=0;i<=n;++i) a[n-i]=s[i]=='*'?0:(s[i]&31);//注意A串取倒串
	for(scanf("%s",s),i=0;i<=m;++i) b[i]=s[i]=='*'?0:(s[i]&31);
	for(P.FFT(n,m,a,b),i=0;i<=m-n;++i) t+=!a[n+i];printf("%d\n",t);//统计个数
	for(i=0;i<=m-n;++i) !a[n+i]&&printf("%d ",i+1);return 0;//输出合法位置
}
posted @ 2019-12-19 16:49  TheLostWeak  阅读(...)  评论(...编辑  收藏