SDUT_3311_數據結構實驗之串三：KMP應用

Problem Description

有n個小朋友，每個小朋友手裏有一些糖塊，現在這些小朋友排成一排，編號是由1到n。現在給出m個數，能不能唯一的確定一對值l和r(l <= r)，使得這m個數剛好是第l個小朋友到第r個小朋友手裏的糖塊數？

Input

首先輸入一個整數n，代表有n個小朋友。下一行輸入n個數，分別代表每個小朋友手裏糖的數量。
之後再輸入一個整數m，代表下面有m個數。下一行輸入這m個數。

Output

如果能唯一的確定一對l,r的值，那麼輸出這兩個值，否則輸出-1

Example Input

5
1 2 3 4 5
3
2 3 4

Example Output

2 4

題意：
找出子串在目標串中的位置，如果在目標串中有多個子串輸出-1；
只有一個時，才輸出所在的位置。
分析：
KMP算法代碼是找到在目標串中有子串就返回其位置，不會繼續往下找，所以要想繼續往下找，就要自己寫嘍。

代碼

#include <iostream>
#include <cstdio>
using namespace std;
int n,m;
int a[100000005],b[100000005];
int book[100000005];
void get_next(int m,int b[])
{
    int i=0;
    int j=-1;
    book[0] = -1;
    while(i < m)
    {
        if(j == -1||b[i] == b[j])
        {
            i++;
            j++;
            if(b[i] != b[j])
                book[i] = j;
            else
                book[i] = book[j];

        }
        else
            j = book[j];
    }
}
void KMP(int a[],int b[])
{
    int i=0;
    int j=0;
    int cnt=0;
    int pos;
    while(i < n && j < m)
    {
        if(j == -1||a[i] == b[j])
        {
            i++;
            j++;
            //一般的模板代碼是找到有子串就返回位置，不會在繼續往後找了
            //所以要想繼續往後走，就要加這一步
            if(j == m)
            {
                cnt++;    //找目標串中有幾個子串，大於一個，輸出-1否則輸出所在位置
                pos = i;   //當前位置(子串最後一個元素所在的位置)
                j = book[j];   //找到一個之後繼續往後找。
                if(cnt > 1)    //多於一個子串就結束循環。
                    break;
            }
        }
        else
            j = book[j];
    }
    if(cnt == 1)
        printf("%d %d",pos-m+1,pos);
    else
        printf("-1\n");
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    scanf("%d",&m);
    for(int j=0;j<m;j++)
        scanf("%d",&b[j]);
    get_next(m,b);
    KMP(a,b);

    return 0;
}