题解 P5685 【[JSOI2013]快乐的 JYY】

皎月半洒花

2019-12-10 16:37:19

Solution

~~PAM为啥总感觉比SAM友善好多啊~~ 与另一篇题解做法不同,我们可以直接对两个串同时建立$\rm PAM$。考虑$\rm PAM$里面每个节点的意义,节点=回文串,所以考虑我们对两个$\rm PAM$同时开始$\sf dfs$,因为起始状态相同,那么如果遇到相同的转移就说明有相同的状态,把$sz_x\times sz_y$作为贡献加到答案里面即可。 以下是一些闲扯: * 注意到,对于回文串之间的转移而言,由于从奇数长度只能转移到奇数长度,偶数长度只能转移到偶数长度,所以要把每一组根都$\sf dfs$一遍。 * 紧接着上一条,显然我们要特判掉虚根(即奇根和偶根)。 * 然后这道题的两个 子$\mathsf {idea}$ 分别跟下面两题相似: * [Link1](https://www.luogu.com.cn/problem/P3649),[Link2](https://www.luogu.com.cn/problem/P5555) 上代码 ```cpp #define Sigma 30 #define MAXN 500010 #define LL long long using namespace std ; int N, M ; LL ans ; char S[MAXN], T[MAXN] ; struct PAM{ int rt0, rt1, last, sz, f[MAXN] ; int trie[MAXN][Sigma], pre[MAXN], len[MAXN] ; void Init(){ sz = -1, rt0 = ++ sz, rt1 = ++ sz ; pre[rt0] = pre[rt1] = rt1, len[rt0] = 0, len[rt1] = -1, last = rt0 ; } void Insert(int x, int p, char *s){ int u = last ; while (s[p] != s[p - len[u] - 1]) u = pre[u] ; if (!trie[u][x]){ int newq = ++ sz, fa = pre[u] ; while (s[p] != s[p - len[fa] - 1]) fa = pre[fa] ; pre[newq] = trie[fa][x], trie[u][x] = newq, len[newq] = len[u] + 2 ; } last = trie[u][x], f[last] ++ ; } }P, Q ; void dfs(int x, int y){ if (x + y > 2) ans += 1ll * P.f[x] * Q.f[y] ; for (int i = 1 ; i <= 26 ; ++ i) if (P.trie[x][i] && Q.trie[y][i]) dfs(P.trie[x][i], Q.trie[y][i]) ; } int main(){ P.Init(), Q.Init() ; cin >> (S + 1) >> (T + 1) ; N = strlen(S + 1), M = strlen(T + 1) ; for (int i = 1 ; i <= N ; ++ i) P.Insert(S[i] - 'A' + 1, i, S) ; for (int i = 1 ; i <= M ; ++ i) Q.Insert(T[i] - 'A' + 1, i, T) ; for (int i = P.sz ; i ; -- i) P.f[P.pre[i]] += P.f[i] ; for (int i = Q.sz ; i ; -- i) Q.f[Q.pre[i]] += Q.f[i] ; dfs(1, 1) ; dfs(0, 0) ; cout << ans<< endl ; return 0 ; } ```