【jzoj4876】【基因突变】【exkmp】

题目大意

707获取了人类的基因信息并尝试对基因进行实验。他发现可以把人类的基因看做一个只包含小写字母的字符串，并定义从头开始任意长度的基因为“源头基因”人类身上与源头基因完全匹配的片段越多，这个人就越容易被控制。于是707就开始了他邪恶的计划……作为人类卫士的射手ZMiG自然不会让707得逞，他决定拯救人类，现在他拿到了其中一个人被改造后的基因，他想请你统计一下它的基因中究竟有多少基因片段是可以与源头基因相匹配的

解题思路

直接exkmp，把求出的数组求和加字符串长的即可。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define LD double
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a>b)?b:a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const inf=1e9;
int const maxn=1e6;
int n,f[maxn+10];
char s[maxn+10];
int main(){
    //freopen("gene.in","r",stdin);
    //freopen("gene.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%s",s+1);
    n=strlen(s+1);
    int id=1;LL ans=n;
    fo(i,2,n){
        if(id+f[id]>i)f[i]=min(id+f[id]-i,f[i-id+1]);
        for(;s[f[i]+1]==s[i+f[i]];f[i]++);
        if(id+f[id]<i+f[i])id=i;
        ans+=f[i];
    }
    printf("%lld",ans);
}