51Nod 1304 字符串的相似度

1304 字符串的相似度

题目来源：

基准时间限制：1 秒 空间限制：131072 KB 分值: 320

给出一个字符串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示例

11

思路：

​ 先说一下什么是扩展kmp算法。

​ 定义母串S，和字串T，设S的长度为n，T的长度为m，求T与S的每一个后缀的最长公共前缀，也就是说，设extend数组,extend[i]表示T与S[i,n-1]的最长公共前缀，要求出所有extend[i](0<=i<n)。

​ 那么这道题就是求出S = s，T = s 的extend，但是要知道扩展kmp算法中Next数组的含义是：next[i]表示T[i,m-1]和T的最长公共前缀长度。注意其实这里就是和extend就类似了，因为如果把extend中的S改为T，就是Next，所以扩展kmp中next和extend求法相同。此题只需求出一次即可。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int maxn=1000005;   //字符串长度最大值  char s[maxn];int Next[maxn]/*,ex[maxn]*/; //ex数组即为extend数组  //预处理计算next数组  void GETNEXT(char *str)  {      int i=0,j,po,len=strlen(str);      Next[0]=len;//初始化next[0]      while(str[i]==str[i+1]&&i+1<len)//计算next[1]             i++;      Next[1]=i;      po=1;       //初始化po的位置      for(i=2;i<len;i++)    {        if(Next[i-po]+i<Next[po]+po)//第一种情况，可以直接得到next[i]的值          Next[i]=Next[i-po];          else//第二种情况，要继续匹配才能得到next[i]的值          {              j=Next[po]+po-i;              if(j<0)j=0;//如果i>po+next[po],则要从头开始匹配              while(i+j<len&&str[j]==str[j+i])//计算next[i]              j++;              Next[i]=j;              po=i;//更新po的位置          }      }  }  /*//计算extend数组  void EXKMP(char *s1,char *s2)  {      int i=0,j,po,len=strlen(s1),l2=strlen(s2);      GETNEXT(s2);//计算子串的next数组      while(s1[i]==s2[i]&&i<l2&&i<len)//计算ex[0]      i++;      ex[0]=i;      po=0;//初始化po的位置      for(i=1;i<len;i++)      {          if(Next[i-po]+i<ex[po]+po)//第一种情况，直接可以得到ex[i]的值          ex[i]=Next[i-po];          else//第二种情况，要继续匹配才能得到ex[i]的值          {              j=ex[po]+po-i;              if(j<0)j=0;//如果i>ex[po]+po则要从头开始匹配              while(i+j<len&&j<l2&&s1[j+i]==s2[j])//计算ex[i]              j++;              ex[i]=j;              po=i;//更新po的位置          }      }  } */int main(int argc, char const *argv[]){    // freopen("in.txt","r",stdin);    scanf("%s",s);    GETNEXT(s);    int ans = 0;    int length = strlen(s);    for(int i = 0;i < length; i++) {        ans += Next[i];    }    printf("%d\n",ans);    return 0;}