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

Hint

#include <iostream>
#include <cstring>
using namespace std;

int a[1000001],s1[1000001],s2[1000001];
int n,m;
void getnext()
{
	int i = 0,j = -1;
	a[0] = -1;
	while(i<m)
	{
		if(j == -1||s2[i] == s2[j])
		{
			i++;
			j++;
			a[i] = j;
		}
		else
			j = a[j];
	}
}
void kmp()
{
	int i = 0 , j = 0,cnt = 0;
	getnext();
	int l,r;
	while(i<n)
	{
		if(j == -1||s1[i] == s2[j])
		{
			++i;
			++j;
		}
		else
			j = a[j];
        if( j == m )
        {
            cnt++;
            if(cnt>2)
            {
                cout<<-1<<endl;
                return ;
            }
            l = i-m+1;
            r = i;
        }
	}
	if(cnt == 1)
        cout<<l<<' '<<r<<endl;
    else
        cout<<-1<<endl;
}
int main(int argc, char const *argv[])
{
	cin>>n;
	for (int i = 0; i < n; ++i)
	{
		cin>>s1[i];
	}
	cin>>m;
	for (int i = 0; i < m; ++i)
	{
		cin>>s2[i];
	}
	kmp();
	return 0;
}