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

p.s.  KMP算法是是一种时间复杂度为O(m+n)的高效字符串查找算法 在时间要求严格的题目中只能打KMP算法

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

Time Limit: 1000 ms Memory Limit: 65536 KiB

Problem 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

Sample Output

1
4
-1

题目链接：

http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/contestproblem/cid/2710/pid/2772

p.s. SDUToj中数组名不能直接定义为next 否则会报错

#include <bits/stdc++.h>

using namespace std;

char s1[1000000],s2[1000000];
int nextnum[1000000];

void get_next(char s2[]){
    int i=0,j=-1,len=strlen(s2);
    nextnum[0]=-1;
    while(i<len)
    {
        if(j==-1||s2[i]==s2[j])
        {
            i++;
            j++;
            nextnum[i]=j;
        }
        else
        j=nextnum[j];
    }
}

int KMP(char s1[],char s2[]){
    int i=0,j=0,len1=strlen(s1),len2=strlen(s2);
    while(i<len1&&j<len2)
    {
        if(j==-1||s1[i]==s2[j])
        {
            i++;
            j++;
        }
        else
        j=nextnum[j];
    }
    if(j==len2)
    return i-j+1;
    else
    return -1;
}

int main()
{
    while(cin >> s1 >> s2)
    {
        get_next(s2);
        cout << KMP(s1,s2) << endl;
    }
    return 0;
}