题意:
求一个不含0的大整数的所有不同回文字串的和
如 : 1121
有 1121、112、121、11、12、21、1、2 八个不同的子串,其中121、11、1、2为回文串
所以其答案为 121 + 11 + 1 + 2 = 135
可以利用回文树得到所有不同的回文串,然后遍历这些回文串,得到所有的回文串的值求和即可。
每个回文串的值有一个递推的关系,可以利用树形DP来得到。
#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
typedef long long int LL;
const int MAXN = 2e6+10;
const int N = 11;
const int mod = 1000000007;
struct Palindromic_Tree {
int next[MAXN][N];//next指针,next指针和字典树类似,指向的串为当前串两端加上同一个字符构成
int fail[MAXN];//fail指针,失配后跳转到fail指针指向的节点
int cnt[MAXN];//cnt[i]表示节点i表示的回文串在原串中出现的个数,要用count
int num[MAXN];//num[i]表示节点i表示的回文串中有几种不同且以该字符结尾的回文串个数
int len[MAXN];//len[i]表示节点i表示的回文串的长度
int S[MAXN];//存放添加的字符
int last;//指向上一个字符所在的节点,方便下一次add
int n;//字符数组指针
int p;//节点指针,节点数
// 0 表示对称轴为空'abba',1 表示对称轴为字符'aba' 起点
int newnode (int l) {//新建节点
for (int i=0; i<N; ++i) next[p][i] = 0;
cnt[p] = 0; num[p] = 0; len[p] = l;
return p ++;
}
void init() {//初始化
p = 0;
newnode(0);
newnode(-1);
last = 0;
n = 0;
S[n] = -1;//开头放一个字符集中没有的字符,减少特判
fail[0] = 1;
}
int get_fail(int x) {//和KMP一样,失配后找一个尽量最长的
while(S[n-len[x]-1]!=S[n]) x = fail[x];
return x;
}
void add(int c) {
c -= '0'; // 字符为 a ~ z
S[++ n] = c;
int cur = get_fail(last);//通过上一个回文串找这个回文串的匹配位置
if (!next[cur][c]) {//如果这个回文串没有出现过,说明出现了一个新的本质不同的回文串
int now = newnode(len[cur] + 2);//新建节点
fail[now] = next[get_fail(fail[cur])][c];//和AC自动机一样建立fail指针,以便失配后跳转
next[cur][c] = now;
num[now] = num[fail[now]] + 1;
}
last = next[cur][c];
cnt[last] ++;
}
void count() {
for(int i=p-1; i>=0; --i) cnt[fail[i]] += cnt[i] ;
//父亲累加儿子的cnt,因为如果fail[v]=u,则u一定是v的子回文串!
}
}_tree;
LL ans = 0, _10[MAXN];
char s[MAXN];
void dfs(int x, int op, LL sum) {
ans = (ans + sum) % mod;
for(int i=1;i<=9;i++) {
int nt = _tree.next[x][i];
if(!nt) continue;
LL to = (sum * 10 % mod + i + i * _10[_tree.len[nt]-1] % mod) % mod;
if(op && _tree.len[nt]==1) to /= 2;
dfs(nt, op, to % mod);
}
}
int main()
{
_10[0] = 1;
scanf("%s", s); int len = strlen(s);
for(int i=1;i<=len;i++) _10[i] = (_10[i-1] * 10) % mod;
_tree.init();
for(int i=0;i<len;i++) _tree.add(s[i]);
dfs(0, 0, 0); dfs(1, 1, 0);
printf("%lld\n", ans % mod);
return 0;
}