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

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

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

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

输入

首先输入一个整数n，代表有n个小朋友。(0

输出

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

示例输入

5
1 2 3 4 5
3
2 3 4

示例输出

2 4

提示

 

来源

windream

此题要唯一确定一对l,r值；理解为确定唯一的子串，即模式串在主串中只出现一次，然后输出匹配模式串在主串中的首位置和末位置

#include<bits/stdc++.h>
using namespace std;

const int Max=1000000;

int next[Max];
int a[Max], b[Max];

void get_next(int len)//求模式串的next数组
{
	int i=0,j=-1;
	next[0]=-1;
	while(i<len)
	{
		if(j==-1||b[i]==b[j])
		{
			i++;
			j++;
			next[i]=j;
		}
		else
		{
			j=next[j];
		}
	}
}

int kmp(int len1, int len2, int x)
{
    get_next(len2);
    int i=x, j=0;
    while(i<len1&&j<len2)
    {
        if(j==-1||a[i]==b[j])
        {
            i++;
            j++;
        }
        else
        {
            j=next[j];
        }
    }
    if(j==len2)
        return i-len2+1;
    else
        return -1;
}

int main()
{
    int n, m;
    while(cin>>n)
    {
        for(int i=0;i<n;i++)
            cin>>a[i];
        cin>>m;
        for(int i=0;i<m;i++)
            cin>>b[i];
        int x=kmp(n, m, 0);
        if(x!=-1)
        {
            int k=kmp(n, m, x+m-1);
            if(k==-1)
                cout<<x<<" "<<x+m-1<<endl;
            else
                cout<<"-1"<<endl;
        }
        else
            cout<<"-1"<<endl;
    }
    return 0;

}