扩展KMP模板(下标从0开始)

拼搏现实的明天。 2021-11-11 09:40 271阅读 0赞

### 题目描述 ###

有两个字符串aa,bb,要求输出bb与aa的每一个后缀的最长公共前缀

### 输入格式 ###

两行,分别为两个字符串aa,bb

### 输出格式 ###

共两行

第一行有lenb个数,为b的next数组(特别地,next\_\{1\}next1为lenb)

第二行有lena个数,即答案

### 输入输出样例 ###

**输入 \#1**复制

aaaabaa
    aaaaa

**输出 \#1**复制

5 4 3 2 1
    4 3 2 1 0 2 1

### 说明/提示 ###

lena<=100000lena<=100000

lenb<=100000lenb<=100000

复杂度  O(N)

#include<bits/stdc++.h>
    
    #define N 1000010 
    
    using namespace std;
    
    int q,nxt[N],extend[N];
    string s,t;
    
    void getnxt()
    {
        nxt[0]=t.size();//nxt[0]一定是T的长度
        int now=0;
        while(t[now]==t[1+now]&&now+1<(int)t.size())now++;//这就是从1开始暴力
        nxt[1]=now;
        int p0=1;
        for(int i=2;i<(int)t.size();i++)
        {
            if(i+nxt[i-p0]<nxt[p0]+p0)nxt[i]=nxt[i-p0];//第一种情况
            else
            {//第二种情况
                int now=nxt[p0]+p0-i;
                now=max(now,0);//这里是为了防止i>p的情况
                while(t[now]==t[i+now]&&i+now<(int)t.size())now++;//暴力
                nxt[i]=now;
                p0=i;//更新p0
            }
        }
    }
    
    void exkmp()
    {
        getnxt();
        int now=0;
        while(s[now]==t[now]&&now<min((int)s.size(),(int)t.size()))now++;//暴力
        extend[0]=now;
        int p0=0;
        for(int i=1;i<(int)s.size();i++)
        {
            if(i+nxt[i-p0]<extend[p0]+p0)extend[i]=nxt[i-p0];//第一种情况
            else
            {//第二种情况
                int now=extend[p0]+p0-i;
                now=max(now,0);//这里是为了防止i>p的情况
                while(t[now]==s[i+now]&&now<(int)t.size()&&now+i<(int)s.size())now++;//暴力
                extend[i]=now;
                p0=i;//更新p0
            }
        }
    }
    
    int main()
    {
        cin>>s>>t;
        exkmp();
        int len=t.size();
        for(int i=0;i<len;i++)printf("%d ",nxt[i]);//输出nxt
        puts("");
        len=s.size();
        for(int i=0;i<len;i++)printf("%d ",extend[i]);//输出extend
        return 0;
    }