Description
兔子们在玩两个串的游戏。给定两个字符串S和T,兔子们想知道T在S中出现了几次, 分别在哪些位置出现。注意T中可能有“?”字符,这个字符可以匹配任何字符。
两行两个字符串,分别代表S和T
Output
第一行一个正整数k,表示T在S中出现了几次 接下来k行正整数,分别代表T每次在S中出现的开始位置。按照从小到大的顺序输出,S下标从0开始。
bbabaababaaaaabaaaaaaaabaaabbbabaaabbabaabbbbabbbbbbabbaabbbababababbbbbbaaabaaabbbbbaabbbaabbbbabab
a?aba?abba
Sample Output
0
HINT
S 长度不超过 10^5, T 长度不会超过 S。 S 中只包含小写字母, T中只包含小写字母和“?”
分析
这题居然不是后缀数组之类的乱搞也是很迷。。用fft来解决就更迷了。。 对于这道题如果没有通配符,我们构造一个函数
f[x]=∑i=1n(s1[i]−s2[i])2
这样如果s1和s2相等当且仅当f[x]=0
但是这道题有通配符,所以我们要把是通配符的位置s2[i]=0
然后变换一下公式可得:
f[x]=∑i=1n(s1[i]−s2[i])2∗s2[i]
这样有通配符的位置相应的函数值就会为0,表示可以匹配一切
但是这样没法做,于是将s2数组拧一拧,翻转过来,发现就是一个卷积之类的了,可用FFT。
代码
#include <bits/stdc++.h>
#define N 300005
#define pi acos(-1)
#define ll long long
using namespace std;
typedef complex<
double> E;
int L,n;
int R[N];
E c[N],d[N],e[N],f[N];
int a[N],b[N];
char s1[N],s2[N];
void fft(E *a,
int f)
{
for (
int i =
0; i < n; i++)
if (i < R[i])
swap(a[i],a[R[i]]);
for (
int i =
1; i < n; i <<=
1)
{
E wn(
cos(pi / i), f *
sin(pi / i));
for (
int j =
0; j < n; j += (i <<
1))
{
E w(
1,
0);
for (
int k =
0; k < i; k++, w *= wn)
{
E x = a[j + k], y = w * a[j + k + i];
a[j + k] = x + y;
a[j + k + i] = x - y;
}
}
}
if (f == -
1)
for (
int i =
0; i < n; i++)
a[i] /= n;
}
int main()
{
scanf(
"%s",s1 +
1);
scanf(
"%s",s2 +
1);
int len1 =
strlen(s1 +
1), len2 =
strlen(s2 +
1);
for (
int i =
1; i <= len1; i++)
a[i] = s1[i] -
'a' +
1;
for (
int i =
1; i <= len2; i++)
b[len2 - i +
1] = s2[i] ==
'?' ?
0 : s2[i] -
'a' +
1;
for (n =
1; n <= len1 *
2; n <<=
1, L++);
for (
int i =
0; i < N; i++)
R[i] = (R[i >>
1] >>
1) | ((i &
1) << (L -
1));
int w =
0;
for (
int i =
1; i <= len2; i++)
w += b[i] * b[i] * b[i];
for (
int i =
1; i <= len1; i++)
c[i] = a[i] * a[i];
for (
int i =
1; i <= len2; i++)
d[i] = b[i];
for (
int i =
1;i <= len1; i++)
e[i] = a[i];
for (
int i =
1; i <= len2; i++)
f[i] = b[i] * b[i] *
2;
fft(c,
1); fft(d,
1); fft(e,
1); fft(f,
1);
for (
int i =
0; i < n; i++)
c[i] = c[i] * d[i], e[i] = e[i] * f[i];
fft(c,-
1); fft(e,-
1);
for (
int i =
0; i < N; i++)
a[i] = (
int)(c[i].real() +
0.1) - (
int)(e[i].real() +
0.1);
int tot =
0;
for (
int i = len2 +
1; i <= len1 +
1; i++)
if (a[i] + w ==
0)
tot++;
cout<<tot<<endl;
for (
int i = len2 +
1; i <= len1 +
1; i++)
if (a[i] + w ==
0)
printf(
"%d\n",i - len2 -
1);
}
转载请注明原文地址: https://ju.6miu.com/read-675240.html