51 nod 1304 1304 字符串的相似度(exkmp)

最新推荐文章于 2022-01-19 16:40:54 发布

mrcoderrev

最新推荐文章于 2022-01-19 16:40:54 发布

阅读量256

点赞数

本文链接：https://blog.csdn.net/yjf3151731373/article/details/77509959

版权

1304 字符串的相似度

题目来源： HackerRank

基准时间限制：1 秒空间限制：131072 KB 分值: 320 难度：7级算法题

关注

我们定义2个字符串的相似度等于两个串的相同前缀的长度。例如 "abc" 同 "abd" 的相似度为2，"aaa" 同 "aaab" 的相似度为3。

给出一个字符串S，计算S同他所有后缀的相似度之和。例如：S = "ababaa"，所有后缀为：

ababaa 6

babaa 0

abaa 3

baa 0

aa 1

a 1

S同所有后缀的相似度的和 = 6 + 0 + 3 + 0 + 1 + 1 = 11

Input

输入一个字符串S(1 <= L <= 1000000)，L为字符串S的长度，且S由a-z的小写字母组成。

Output

输出S同所有后缀的相似度的和。

Input示例

ababaa

Output示例

#include <iostream>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6+10;
const LL mod = 1e9+7;
char s1[N], s2[N];
int  nex[N], nex1[N];
void getnext(char a[],int len)
{
    int k=0, i=1;
    nex[0]=len;
    while(k+1<len&&a[k]==a[k+1])++k;
    nex[1]=k;
    k=1;
    while(++i<len)
    {
        int p=k+nex[k]-1;
        nex[i]=min(nex[i-k],max(p-i+1,0));
        while(i+nex[i]<len&&a[nex[i]]==a[i+nex[i]]) ++nex[i];
        if(i+nex[i]>k+nex[k]) k=i;
    }
    return ;
}


int main()
{
    scanf("%s",s1);
    int len=strlen(s1);
    getnext(s1,len);
    LL ans=0;
    for(int i=0;i<len;i++)
    {
        ans=ans+(LL)nex[i];
    }
    printf("%lld\n",ans);
    return 0;
}