[luogu4173]残缺的字符串
求对于(B)的每一个位置 (i),从这个位置开始连续(m)个字符形成的子串是否可能与(A)串完全匹配.(1le mle n le 3*{10}^5)
两个位置(A_i)与(B_{j+i})匹配有三种情况:
- (A_i)为'*'
- (B_{i+j})为'*'
- (A_i = B_{i+j}),即(A_i-B_{i+j}=0)
那么显然可以令''所在位置为0,把三种情况乘起来就可以判断一个位置是否匹配了
判断从(j)开始的(m)个位置是否匹配就是$$S(j)=sum_{i=1}m(A_i-B_{i+j})2A_i*B_{i+j}$$
[S(j)=sum_{i=1}^m(A_i^2-2*A_i*B_{i+j}+B_{i+j}^2)*A_i*B_{i+j}
]
[=sum_{i=1}^mA_i^3*B_{i+j}-2sum_{i=1}^mA_i^2*B_{i+j}^2+sum_{i=1}^mA_i*B_{i+j}^3
]
把(A)翻转求卷积就好了
vector<int> q;
double eps=1e-6;
int dcmp(double x){if(fabs(x) < eps) return 0; return 1;}
char c[Maxn];
void solve(){
m=read(), n=read(); scanf("%s", c+1);
for(int i=1; i <= m; i++) f[1][m-i].a=c[i] == '*' ? 0 : c[i]-'a'+1,
f[2][m-i].a=f[1][m-i].a*f[1][m-i].a, f[3][m-i].a=f[1][m-i].a*f[2][m-i].a;
scanf("%s", c);
for(int i=0; i < n; i++) g[1][i].a=c[i] == '*' ? 0 : c[i]-'a'+1,
g[2][i].a=g[1][i].a*g[1][i].a, g[3][i].a=g[1][i].a*g[2][i].a;
while(lim <= n+m-2) lim<<=1, l++;
for(int i=0; i < lim; i++) r[i]=(r[i>>1]>>1)|((i&1)<<l-1);
for(int i=1; i <= 3; i++) FFT(f[i], 1), FFT(g[i], 1);
for(int i=1; i <= 3; i++) for(int j=0; j < lim; j++) f[i][j]=f[i][j]*g[4-i][j];
for(int i=1; i <= 3; i++) FFT(f[i], -1);
double delta;
for(int i=m-1; i < n; i++) {
delta=(f[1][i].a-2*f[2][i].a+f[3][i].a)/lim;
if(!dcmp(delta)) q.push_back(i-m+2); //printf("%d %lf
", i, delta);
}
printf("%d
", q.size());
for(int i=0; i < q.size(); i++) printf("%d ", q[i]);
}