题目描述
给定一个模式串S,以及一个模板串P,所有字符串中只包含大小写英文字母以及阿拉伯数字。
模板串P在模式串S中多次作为子串出现。
求出模板串P在模式串S中所有出现的位置的起始下标。
输入格式
第一行输入整数N,表示字符串P的长度。
第二行输入字符串P。
第三行输入整数M,表示字符串S的长度。
第四行输入字符串S。
输出格式
共一行,输出所有出现位置的起始下标(下标从0开始计数),整数之间用空格隔开。
数据范围
1≤N≤104
1≤M≤105
样例
输入样例:
3
aba
5
ababa
输出样例:
0 2
算法1
(暴力枚举) $O(n^2)$
C++ 代码
#include <bits/stdc++.h>
using namespace std
void kmp(const string& p, const string& s) {
const int m = p.size();
const int n = s.size();
for(int i = 0; i < n; i++) {
bool flag = true;
for(int j = 0; j < m; j++) {
if(s[i + j] == p[j])
continue;
flag = false;
break;
}
if(flag) cout << i << ' ';
}
}
int main()
{
int n,m;
string p,s;
cin>>n>>p>>m>>s;
kmp(p,s);
return 0;
}
算法2
(kmp) $O(n)$
C++ 代码
#include <bits/stdc++.h>
using namespace std;
const int N=1e4;
const int M=1e5;
int n,m;
char s[M],p[N];
int ne[N];
int main()
{
cin>>n>>p+1>>m>>s+1;
for(int i=2,j=0;i<=n;i++)//ne[1]=0
{
while(j&&p[i]!=p[j+1]) j=ne[j];
if(p[i]==p[j+1]) j++;
ne[i]=j;
}
for(int i=1,j=0;i<=m;i++)
{
while(j&&s[i]!=p[j+1]) j=ne[j];
if(s[i]==p[j+1]) j++;
if(j==n)
{
j=ne[j];
printf("%d ",i-n);
}
}
return 0;
}