题意简述
一个字符串是“优秀”的,定义为:存在两个非空的字符串 A , B A,B A,B,满足这个字符串 S = A A B B S=AABB S=AABB,那么这个 S S S 就是优秀的。
给定一个串,计算它有多少子串是“优秀”的。相同的子串出现在不同的位置也被算两次。
长度 ≤ 3 e 4 \le 3e4 ≤3e4
思路
对于每个位置,我们维护 p r e [ i ] pre[i] pre[i] 表示以 i i i 结尾的 A A AA AA 形式的串有多少个, s u f [ i ] suf[i] suf[i] 同理。
然后答案就是 ∏ i = 1 n − 1 p r e [ i ] × s u f [ i + 1 ] \prod\limits_{i=1}^{n-1}pre[i]\times suf[i+1] i=1∏n−1pre[i]×suf[i+1]
然后考虑怎么求 p r e pre pre。( s u f suf suf 的话,就把串反过来求一遍 p r e pre pre 即可)
如何求 pre
不会吧?不会就对了,黑题是这么容易做出来的嘛。 我们继续降低要求,我们求特定长度的 p r e [ i ] pre[i] pre[i],比如 A A A 的长度为 k k k,也就是 A A AA AA 串的长度为 2 k 2k 2k 。 这个好判断,哈希就行了。 对于位置 i i i,如果 [ i − k + 1 , i ] [i-k+1,i] [i−k+1,i] 区间的哈希值等于区间