数据结构实验之串一：KMP简单应用

数据结构实验之串一：KMP简单应用

Time Limit: 1000MS Memory Limit: 65536KB

Submit Statistic

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

Hint

 

Author

cjx 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
char a[1000000+5], b[1000000+5];
int next[1000000+5];
void getnext() {
    next[0]=-1;
    int i=0, j=-1;
    while(b[i]!='\0') {
        if(j==-1||b[i]==b[j]) {
            i++;
            j++;
            next[i]=j;
        }
        else j=next[j];
    }
}
void kmp() {
    getnext();
    int i=0, j=0;
    int n=strlen(a), m=strlen(b);
    while(i<n&&j<m) {
        if(a[i]==b[j]||j==-1) {
            i++;
            j++;
        }
        else j=next[j];
    }
    if(j>=m) printf("%d\n", i-m+1);
    else printf("-1\n");
}
int main() {
    while(~scanf("%s %s", a, b)) {
        kmp();
    }
    return 0;
}