OJ3311數據結構實驗之串三：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

示例程序

注意題目中的是否存在唯一確定的一對

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define max 1000000
int next[max];
int a[max],b[max];
typedef struct
{
    int *ch;
    int len;
} String;

void init(String *tr)
{
    tr->ch=(int *)malloc(max*sizeof(int));
    if(!tr->ch) return ;
    tr->len=0;
}

void creat(String *str,int a[],int n)
{
    int i;
    for(i=0; i<n; i++)
        str->ch[i+1]=a[i];
    str->len=i;
}

void get_nextval(String *str)
{
    int i=1,j=0;
    next[1]=0;
    while(i<str->len)
    {
        if(j==0||str->ch[i]==str->ch[j])
        {
            ++i;
            ++j;
            next[i]=j;
        }
       else j=next[j];
    }
}

int KMP(String *str1,String *str2)
{
    int i=1,j=1,f=0,t;
    get_nextval(str2);
    while(i<=str1->len&&j<=str2->len)
    {
        if(j==0||str1->ch[i]==str2->ch[j])
        {
            ++i;
            ++j;
        }
        else
            j=next[j];
        if(j>str2->len)
        {
            t=i;
            j=1;
            f++;
        }
    }
    if(f==1) return t-str2->len;
    else return -1;
}

int main()
{
    String str1,str2;

    int n,i,m,x;
    init(&str1);
    init(&str2);
    scanf("%d",&n);
    for(i=0; i<n; i++)
    {
        scanf("%d",&a[i]);
    }
    scanf("%d",&m);
    for(i=0; i<m; i++)
    {
        scanf("%d",&b[i]);
    }
    creat(&str1,a,n);
    creat(&str2,b,m);
    int f=KMP(&str1,&str2);
    if(f!=-1)
    {
        printf("%d %d\n",f,f+m-1);
    }

    else
        printf("%d\n",f);
}