题目:给串s和串t,求从s中取一个子串,再和t的一个前缀拼接起来能组成回文串的个数,要求从s取的子串比取的t的前缀长。
可以用扩展kmp+回文算法做,也可以用sam替代扩展kmp,但是很难写。
回文算法用马拉车和回文树都可以,但是这里要求的是“从当前位置开始有多少回文串”,马拉车的数组要经过处理才能用,回文树就可以直接用了…
果然回文树各种方面都完爆马拉车,只是有点长。
可以用扩展kmp求出s串的反串每个后缀与t串的最长公共前缀,然后用每个lcp * 下一个位置的回文个数,累加即是答案。
sam就比较麻烦了,一开始思考倒着建s串的sam,但是倒着建无法很好的在自动机上更新出每个位置的答案,只好正着建,然后倒着跑t串,跑到最后再跳link去统计答案。
然而有一个注意的地方就是跑到最后位置的匹配长度要单独记录,因为这个匹配长度可能不是自动机当前节点记录的最长长度,向上跳的时候就没问题了,直接求出当前节点代表的子串个数,再乘当前节点更新出来的回文个数,累加就是答案。
Sam + Pam的ac代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
char s[maxn], t[maxn];
//回文树+后缀自动机解法
struct Pam {
int next[maxn][26];
int fail[maxn];
int len[maxn];// 当前节点表示回文串的长度
int num[maxn];// 到当前节点这里有多少本质不同的回文子串
int pa[maxn];
int S[maxn];
int last, n, p;
int newNode(int l) {
memset(next[p], 0, sizeof(next[p]));
len[p] = l;
num[p] = 0;
return p++;
}
void init() {
n = last = p = 0;
newNode(0);
newNode(-1);
S[n] = -1;
fail[0] = 1;
}
int getFail(int x) {
while(S[n - len[x] - 1] != S[n]) {
x = fail[x];
}
return x;
}
int add(int c) {
S[++n] = c;
int cur = getFail(last);
if(!next[cur][c]) {
int now = newNode(len[cur] + 2);
fail[now] = next[getFail(fail[cur])][c];
next[cur][c] = now;
num[now] = num[fail[now]] + 1;
}
last = next[cur][c];
return num[last];
}
void build() {
init();
int len = strlen(s);
for(int i = len - 1; i >= 0; i--) {
pa[i] = add(s[i] - 'a');
}
// for(int i = 0; i < len; i++) {
// printf("%d ", pa[i]);
// }
// printf("\n");
}
} pa
串处理与回文串算法

最低0.47元/天 解锁文章
1157





