SDUT OJ 数据结构实验之串三：KMP应用

数据结构实验之串三：KMP应用

Time Limit: 1000 ms Memory Limit: 65536 KiB

Submit Statistic Discuss

Problem Description

有n个小朋友，每个小朋友手里有一些糖块，现在这些小朋友排成一排，编号是由1到n。现在给出m个数，能不能唯一的确定一对值l和r(l <= r)，使得这m个数刚好是第l个小朋友到第r个小朋友手里的糖块数？

Input

首先输入一个整数n，代表有n个小朋友。下一行输入n个数，分别代表每个小朋友手里糖的数量。

之后再输入一个整数m，代表下面有m个数。下一行输入这m个数。

Output

如果能唯一的确定一对l,r的值，那么输出这两个值，否则输出-1

Sample Input

5

1 2 3 4 5

3

2 3 4

Sample Output

2 4

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

int s1[1000005],s2[1000005];

int next[1000005];

void get_next(int *s, int N)

{

    next[0] = -1;

    int k = -1;

    for (int i=1; i<=N-1; i++)

    {

        while (k>-1 && s[k+1] != s[i])

            k = next[k];

        if (s[k+1] == s[i])

            k++;

        next[i] = k;

    }

}

void KMP( int *s, int *p, int n, int m )

{

    int l, r, i;

    int cnt = 0;

    int k = -1;

    for ( i = 0; i < n; i++)

    {

        while (k >-1&& p[k + 1] != s[i])

            k = next[k];

        if (p[k + 1] == s[i])

            k++;

        if (k == m-1)

        {

            l = i - m + 2;

            r = i + 1;

            cnt++;

        }

    }

    if( cnt == 1 )

        printf("%d %d\n", l, r );

    else printf("-1\n");

    return ;

}

int main()

{

    int n, m, i;

    scanf("%d", &n);

    for(i=0; i<n; i++)

        scanf("%d", &s1[i]);

    scanf("%d", &m);

    for(i=0; i<m; i++)

        scanf("%d", &s2[i]);

    get_next(s2, m);

    KMP(s1, s2, n, m);

    return 0;

}