2022牛客寒假算法基础集训营6 回文大师 kmp

最新推荐文章于 2022-03-25 19:06:53 发布

不吃土司边

最新推荐文章于 2022-03-25 19:06:53 发布

阅读量343

点赞数

分类专栏： acm 文章标签：算法 c++

本文链接：https://blog.csdn.net/weixin_45714303/article/details/123000861

版权

KMP算法字符串匹配子串计数 long long C++

关键词由CSDN通过智能技术生成

acm 专栏收录该内容

302 篇文章 1 订阅

订阅专栏

#include<bits/stdc++.h>
using namespace std;
#define int long long 
const int N=2e6+10;
vector<int> nxt;
int s[N],ans[N],val[N],p[N];
int n;
vector<int> getnxt() {
  vector<int> pi(n);
  for (int i = 1; i < n; i++) {
    int j = pi[i - 1];
    while (j > 0 && p[i] != p[j]) j = pi[j - 1];
    if (p[i] == p[j]) j++;
    pi[i] = j;
  }
  return pi;
}
void KMP_Count() {
    for(int i = 0, j = 0; i < n; ++i){
        while(j && s[i] != p[j]) j = nxt[j-1];
        if(s[i] == p[j]) j++;
        val[j]++;
    }
}
int32_t main(){
    scanf("%lld",&n);
    for(int i=0;i<n;i++) scanf("%lld",&p[i]);
    for(int i=0;i<n;i++) s[i]=p[n-i-1];
    nxt=getnxt();KMP_Count();
    for(int i=n;i>=1;i--) val[nxt[i-1]]+=val[i];
    for(int i=1;i<=n;i++){
        cout<<val[i]<<" ";
    } cout<<endl;

}