Description
给定两个字符串string1和string2,判断string2是否为string1的子串。
Input
输入包含多组数据,每组测试数据包含两行,第一行代表string1(长度小于1000000),第二行代表string2(长度小于1000000),string1和string2中保证不出现空格。
Output
对于每组输入数据,若string2是string1的子串,则输出string2在string1中的位置,若不是,输出-1。
Sample
Input
abc
a
123456
45
abc
ddd
Output
1
4
-1
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
void get_next(char t[],int next[])
{
int i = 0;
int j = -1;
next[0] = -1;
while(t[i] != '\0')
{
if(j == -1 || t[j] == t[i])
{
i++;
j++;
next[i] = j;
}
else
j = next[j];
}
}
int kmp(char s[],char t[])
{
int slen = strlen(s);
int tlen = strlen(t);
int next[tlen];
int i = 0, j = 0;
get_next(t,next);
while(i<slen && j<tlen)
{
if(j == -1 || s[i] == t[j])
{
j++;
i++;
}
else
j = next[j];
}
if(j >= tlen)
return i-tlen+1;
else
return -1;
}
int main()
{
ios::sync_with_stdio(false);
char s[1000005],t[1000005];
while(cin>>s>>t)
cout<<kmp(s,t)<<endl;
return 0;
}