数据结构实验之串一:KMP简单应用
Time Limit: 1000MS Memory Limit: 65536KB
Problem Description
给定两个字符串string1和string2,判断string2是否为string1的子串。
Input
输入包含多组数据,每组测试数据包含两行,第一行代表string1(长度小于1000000),第二行代表string2(长度小于1000000),string1和string2中保证不出现空格。
Output
对于每组输入数据,若string2是string1的子串,则输出string2在string1中的位置,若不是,输出-1。
Example Input
abc
a
123456
45
abc
ddd
Example Output
1
4
-1
#include<stdio.h>
#include<string.h>
char str1[1000010], str2[1000010];
int next[1000010];
void get_next(char s[])
{
int i, j, len = strlen(s);
next[0] = 0;
for(i = 1; i < len; i++)
{
j = next[i-1];
while(j > 0 && s[j] != s[i])
{
j = next[j - 1];
}
if(s[i] == s[j])
{
next[i] = j + 1;
}
else
{
next[i] = 0;
}
}
}
int KMP(char str1[], char str2[])
{
int i = 0, j = 0;
while(i < strlen(str1) && j < strlen(str2))
{
if(str1[i] == str2[j])
{
i++;
j++;
}
else
{
if(j == 0)
{
i++;
}
else
{
j = next[j - 1];
}
}
}
if(j >= strlen(str2)) return i - strlen(str2) + 1;
else return -1;
}
int main()
{
while(~scanf("%s %s", str1, str2))
{
get_next(str2);
printf("%d\n", KMP(str1, str2));
}
return 0;
}