码迷,mamicode.com
首页 > 其他好文 > 详细

Number Sequence(KMP,判断子串 模板)

时间:2015-08-16 17:57:34      阅读:92      评论:0      收藏:0      [点我收藏+]

标签:

题意:

给两数组,求一个是否是另一个的子数组,若是返回匹配的首位置

分析:

KMP 入门

//扫描字符串A,并更新可以匹配到B的什么位置。
#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
int a[1000001],b[10001],next[10001];
void getNext(int m){
    next[0]=-1;
    int i=0,j=-1;
    while(i<m){
        if(j==-1||b[i]==b[j]){
            i++;
            j++;
            if(b[i]==b[j])next[i]=next[j];
            else next[i]=j;
        }
        else j=next[j];
    }
}
void KMP(int n,int m){
    int i=0,j=0;
    while(i<n&&j<m){
        if(a[i]==b[j]||j==-1){
            i++;
            j++;
        }
        else j=next[j];
    }
    if(j==m){
        printf("%d\n",i-m+1);
    }
    else printf("-1\n");
}
int main()
{
    int n,m,t;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;++i)
            scanf("%d",&a[i]);
        for(int j=0;j<m;++j)
            scanf("%d",&b[j]);
        if(m>n)printf("-1\n");
        else {
            getNext(m);
            KMP(n,m);
        }
    }
return 0;
}

 

Number Sequence(KMP,判断子串 模板)

标签:

原文地址:http://www.cnblogs.com/zsf123/p/4734545.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!