KMP-Number Sequence-HDU - 1711/Acwing-KMP字符串

KMP-Number Sequence-HDU - 1711/Acwing-KMP字符串

一：Number Sequence-HDU - 1711

题目：

给定两个数字序列 a[] 和 b[]，b[] 有可能整体作为一个连续子序列出现在了 a[] 中，现在请你找出 b[] 在 a[] 中第一次出现的位置（起始位置从 1 开始计数），如果一次都没有出现，请输出 -1。

输入格式

第一行包含一个数字 T，表示测试用例的个数。
对于每组测试用例，第一行包含两个数字 n m ( 1<= n <= 1000000, 1 <= m <= 10000 )，表示 a[] 和 b[] 的长度。
接下来的一行包括 n 个数字，依次表示 a[1], a[2], a[3] … a[n]，-1000000 <= a[i] <= 1000000
接下来的一行包括 m 个数字，依次表示 b[1], b[2], b[3] … b[m]，-1000000 <= b[i] <= 1000000

输出格式

对于每组数据输出一行，请输出 b[] 整体作为一个连续子序列在 a[] 中的首次出现位置的起始下标，如果一次都没有完整出现，输出 -1。

样例输入

2
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 2 1

样例输出

6
-1

样例解释

对于第一组数据，1 2 1 2 3 [1 2 3 1 3] 2 1 2，首次出现位置的起始下标为 6。
对于第二组数据，b[] 没有出现在 a[] 中，输出 -1。

---

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<unordered_map>
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e6+10;
int T,n,m;
int a[N],b[N];
int Next[N];

void get_next()
{
    for(int i=2,j=0;i<=m;i++)
    {
        while(j&&b[i]!=b[j+1]) j=Next[j];
        if(b[i]==b[j+1]) j++;
        Next[i]=j;
    }
}

int main()
{
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=1;i<=m;i++) scanf("%d",&b[i]);

        get_next();

        bool flag=true;
        for(int i=1,j=0;i<=n;i++)
        {
            while(j&&a[i]!=b[j+1]) j=Next[j];
            if(a[i]==b[j+1]) j++;

            if(j==m)
            {
                printf("%d",i-m+1);
                flag=false;
                break;
            }
        }
        if(flag) puts("-1");
        else puts("");
    }

    return 0;
}

---

二：KMP字符串

题目：

给定一个模式串S，以及一个模板串P，所有字符串中只包含大小写英文字母以及阿拉伯数字。

模板串P在模式串S中多次作为子串出现。

求出模板串P在模式串S中所有出现的位置的起始下标。

输入格式
第一行输入整数N，表示字符串P的长度。

第二行输入字符串P。

第三行输入整数M，表示字符串S的长度。

第四行输入字符串S。

输出格式
共一行，输出所有出现位置的起始下标（下标从0开始计数），整数之间用空格隔开。

数据范围
1≤N≤104
1≤M≤105
输入样例：
3
aba
5
ababa
输出样例：
0 2

---

KMP算法模板：

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

const int N=1e5+10;
int Next[N];   ///Next[i]前i-1位最长的公共前后缀的长度
int n,m;		///n是模式字符串p的长度，m是原字符串s的长度
char p[N],s[N];

void get_table()
{
    for(int i=2,j=0;i<=n;i++)   ///字符串下标从1开始，这里从2开始
    {
        while(j&&p[i]!=p[j+1]) j=Next[j];
        if(p[i]==p[j+1])
            j++;
        Next[i]=j;
    }
}

int main()
{

    cin>>n>>p+1>>m>>s+1;
    
    get_table();
    
    for(int i=1,j=0;i<=m;i++)
    {
        while(j&&s[i]!=p[j+1]) j=Next[j];
        if(s[i]==p[j+1]) j++;
        
        if(j==n)   ///模式字符串匹配到尾部，说明匹配成功
        {
            j=Next[j];  ///回头继续，可能还有其他子串
            printf("%d ",i-n);   ///原字符串指针减去模式字符串长度即其在原字符串中的起始位置
        }
    }
    
    return 0;
}

---

STL find函数的使用

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

int n,m;
string p,s;

int main()
{

    cin>>n>>p>>m>>s;
    
    int i=0;
    while(s.find(p,i)!=-1)   ///查找失败返回-1
    {
        cout<<s.find(p,i)<<" ";  ///从第i个位置起查找子串p
        i=s.find(p,i)+1;
    }
    
    return 0;
}