#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MXN = 1e5 + 5;
const int p = 13131;
const int mod = 1e9 + 7;
int n, has[MXN], pn[MXN];
char s[MXN];
int get_hash(int l, int r) {
int ans = (has[r] - (long long)has[l-1]*pn[r-l+1]) % mod;
return (ans + mod) % mod;
}
int main() {
pn[0] = 1;
for(int i = 1; i < MXN; ++i) pn[i] = (long long)pn[i-1] * p % mod;
scanf("%s", s + 1);
n = strlen(s + 1);
for(int i = 1; i <= n; ++i) {
has[i] = ((long long)has[i - 1] * p + s[i] - 'a' + 1) % mod;
}
long long ans = n;
for(int i = 2; i <= n; ++i) {
int L = 1, R = n - i + 1, mid, res = 0;
while(L <= R) {
mid = (L + R) >> 1;
if(get_hash(1, mid) == get_hash(i, i + mid - 1)) {
res = mid;
L = mid + 1;
}else {
R = mid - 1;
}
}
ans += res;
}
printf("%lld\n", ans);
return 0;
}
题目描述
小明刚学习完最长公共前缀(zhui),觉得很有趣。
前缀就是原字符串从左往右的一段连续子字符串。abc的前缀是a, ab, abc。
两个字符串的最长公共前缀就是两个字符串从左往右最长的一段相同的连续子字符串。abc和abd的最长公共前缀是ab。
给小明一个长度为nn的字符串,请求出字符串的每个后缀与其本身的最长公共前缀的长度之和。
输入
一行一个字符串S。 1=<|S|<=100000。
输出
一行一个整数表示答案。
样例输入 复制
aaa
样例输出 复制
6
提示
样例输入2
qqwqe
样例输出2
7