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

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

Time Limit: 1000MS Memory limit: 65536K

题目描述

给定两个字符串string1和string2，判断string2是否为string1的子串。

输入

 输入包含多组数据，每组测试数据包含两行，第一行代表string1(长度小于1000000)，第二行代表string2（长度小于1000000），string1和string2中保证不出现空格。

输出

 对于每组输入数据，若string2是string1的子串，则输出string2在string1中的位置，若不是，输出-1。

示例输入

abc
a
123456
45
abc
ddd

示例输出

1
4
-1

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef char status;
typedef struct
{
    char *ch;
    int len;
}str;
status inits(str &s)
{
    s.ch = (char *)malloc(sizeof(char));
    if(!s.ch) exit(0);
    s.ch = NULL;
    s.len = 0;
    return 1;
}

status assigns(str &s, char *ss)
{
    if(s.ch)
        free(s.ch);
    int i=0;
    while(ss[i]!='\0')
        i++;
    if(!i)
    {
        s.ch = NULL;
        s.len = 0;
    }
    else
    {
        s.ch = (char *)malloc(i * sizeof(char));
        if(!s.ch)  exit (0);
        int j=0;
        for(j=0;j<i;j++)
            s.ch[j]=ss[j];
        s.len = i;
    }
    return 1;
}

int strlen(str &s)
{
    return s.len;
}
int next[1000010];
void getnext(str &s2)
{
    int i=0;
    next[0]=-1;
    int j=-1;
    while(i<s2.len)//自己跟自己匹配
    {
        if(j==-1||s2.ch[i]==s2.ch[j])
        {
            i++;
            j++;
            next[i]=j;  
        }
        else
            j=next[j];
    }
}

int index_kmp(str &s1, str &s2)
{
    getnext(s2);
    int i=0, j=0;
    while(i<s1.len&&j<s2.len)
    {
        if(j==-1||s1.ch[i]==s2.ch[j])//next[0]=-1||不失配
        {
            i++;
            j++;
        }    //继续比较后序字符
        else //若从除0之外的位置失配，s1的i指针不回溯， 从s2的k位置开始匹配
            j=next[j];
    }
    if(j>=s2.len) return i-s2.len+1;//子串结束， 说明匹配成功
    else
        return -1;
}

char str1[1000010], str2[1000010];
int main()
{
    str s1, s2;
    while(~scanf("%s", str1))
    {
        scanf("%s", str2);
        inits(s1);
        inits(s2);
        assigns(s1, str1);
        assigns(s2, str2);
        printf("%d\n", index_kmp(s1, s2));
    }
}