【模板】KMP字符串匹配
#include <bits/stdc++.h>
using namespace std;
#define MAXN 1000000
int next[MAXN];
void get_next (string T,int *next){ //得到next数组
int j = 0;
next[0] = 0;
int i = 1;
for(i = 1;i < T.size();i ++){
while(j > 0 && T[i] != T[j]){
j = next[j - 1];
}
if(T[i] == T[j]){
j ++;
}
next[i] = j;
}
}
int main ()
{
int next[MAXN];
string S;
string T;
cin >> S;
//getchar();
cin >> T;
// getchar();
get_next(T, next);
int i = 0;
int j = 1;
while(i <= S.size() && j <= T.size()){
if( j == 0 || S[i] == T[j]){
if(j == (T.size() - 1)){ //判断是否匹配完
//cout << i - T.size() + 2 << endl;
printf("%d\n",(i - T.size() + 2));
}
i ++;
j ++;
}
else {
j = next[j - 1]; //返回上一个next的值
}
}
for(int i = 0;i < T.size() - 1;i ++){ //输出每个next的值
cout << next[i] << ' ';
}
cout << next[T.size() - 1];
return 0;
}
Compress Words
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1000000+66;
const ll mod=1e9+7;
int n;
string s[maxn];
string ans;
int nexts[maxn];
void getnexts(string t) //求出每个字符串的next的值
{
int len=t.size();
int i=0;
int j=-1;
nexts[0]=-1;
while(i<len)
{
if(j==-1||t[i]==t[j])
{
i++;
++j;
if(t[i]!=t[j])
nexts[i]=j;
else
nexts[i]=nexts[j];
}
else
{
j=nexts[j];
}
}
}
int kmp(string &s,string &t,int pos)
{
int i=pos;//
int j=0;
int sl=s.size();
int tl=t.size();
while(i<sl)
{
if(j==-1||s[i]==t[j])
{
++i;
++j;
}
else
{
j=nexts[j];
}
}
return j;
}
void add(string s) //将相邻字符串连接在一起
{
getnexts(s);
int pos=kmp(ans,s,max(0,(int)ans.size()-(int)s.size()));
for(int i=pos; i<s.size(); i++)
{
ans.push_back(s[i]);
}
}
int main()
{
scanf("%d",&n);
for(int i=1; i<=n; i++)
{
cin>>s[i];
}
ans=s[1];
for(int i=2; i<=n; i++)
{
add(s[i]);
}
cout<<ans;
}