1. 程式人生 > >洛谷 P3375 【模板】KMP字符串匹配

洛谷 P3375 【模板】KMP字符串匹配

clu str ext 字符 using main 出現 前綴 mes

題目描述

如題,給出兩個字符串s1和s2,其中s2為s1的子串,求出s2在s1中所有出現的位置。
輸入輸出格式
輸入格式:
第一行為一個字符串,即為s1
第二行為一個字符串,即為s2
輸出格式:
若幹行,每行包含一個整數,表示s2在s1中出現的位置
接下來1行,包括length(s2)個整數,表示前綴數組next[i]的值。

p[i]存指針,指向字符串失配後再次匹配的位置

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000000+100;
int p[maxn];
char a[maxn],b[maxn];
int n,m;
void pre()//求p[i]相當於對自己串求kmp
{
    p[1]=0;
    int j=0;
    for(int i=1;i<m;i++)
      {
      while(j>0&&b[j+1]!=b[i+1])j=p[j];
      if(b[j+1]==b[i+1])++j;
      p[i+1]=j;
      }
      
}
void kmp()
{
    int j=0;
    for(int i=0;i<n;i++)
    {
        while(j>0&&a[i+1]!=b[j+1])j=p[j];
        if(a[i+1]==b[j+1])++j;
        if(j==m)
        {
            printf("%d\n",i+2-m);
            j=p[j];//繼續匹配
        }
    }
}
int main()
{
    scanf("%s%s",a+1,b+1);
    n=strlen(a+1),m=strlen(b+1);
    pre();
    kmp();
    for(int i=1;i<=m;i++)
    printf("%d ",p[i]);
    return 0;
}

洛谷 P3375 【模板】KMP字符串匹配