sdut_3311_数据结构实验之串三：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

Hint

Source

windream

重点是判断是否是唯一的一组解

代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int st[1000004];
int str[1000004];
int next[1000004];

void Find_next(int l)
{
    int i = 0;
    int j = -1;
    next[0] = -1;
    while(i < l)
    {
        if(j == -1 || str[i] == str[j])
        {
            i++;
            j++;
            next[i] = j;
        }
        else
            j = next[j];
    }
}

void kmp(int l, int s)
{
    int i = 0, j = 0;
    int num = 0;
    int k;                                       // 用k记录下标
    while(i < s)                             //去掉了j < s   判断是否只有一组
    {
        if(j == -1 || st[i] == str[j])
        {
            i++;
            j++;
            if(j == l)                         //当有一组时，num++
            {
                num++;
                if(num == 1)
                    k = i;
                else if(num > 1)      // 大于一组直接break
                {
                    break;
                }
                j = 0;                      //否则重新开始
                i = i - l + 1;             //   i 是 不从0开始 而是在判断完一组之后的一个数字开始
            }
        }
        else
            j = next[j];
    }
    if(num == 1)
        printf("%d %d\n", k - l + 1, k);
    else
        printf("-1\n");
}

int main()
{
    int n, m;
    int i;
    scanf("%d", &n);
    for(i = 0; i < n; i++)
        scanf("%d", &st[i]);
    scanf("%d", &m);
    for(i = 0; i < m; i++)
        scanf("%d", &str[i]);
    Find_next(m);
    kmp(m, n);
    return 0;
}