?后綴自動機+dp

一個串在另一個串上跑。

先對A建出自動機，然后用B在上面跑，記錄當前匹配的最大長度，每次經過一個節點記錄經過次數，并加上(len-Max(par))*Right，是這個狀態對答案的貢獻，然后把每個節點的出現次數向par樹上的祖先推一遍計算貢獻。

#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N = 4e5 + 5; int n, m; ll ans; ll sum[N], Right[N], apr[N], f[N]; int a[N], c[N]; char s1[N], s2[N]; namespace SAM {struct node {int val, par;int ch[26];} t[N];int last = 1, root = 1, sz = 1;int nw(int x){t[++sz].val = x;return sz;}void extend(int c){int p = last, np = nw(t[p].val + 1);while(p && !t[p].ch[c]) t[p].ch[c] = np, p = t[p].par;if(!p) t[np].par = root;else{int q = t[p].ch[c];if(t[q].val == t[p].val + 1) t[np].par = q;else{int nq = nw(t[p].val + 1);memcpy(t[nq].ch, t[q].ch, sizeof(t[q].ch));t[nq].par = t[q].par;t[q].par = t[np].par = nq;while(p && t[p].ch[c] == q) t[p].ch[c] = nq, p = t[p].par;}}Right[np] = 1;last = np;} } using namespace SAM; int main() {scanf("%s%s", s1 + 1, s2 + 1);n = strlen(s1 + 1);m = strlen(s2 + 1);for(int i = 1; i <= n; ++i) extend(s1[i] - 'a');for(int i = 1; i <= sz; ++i) ++c[t[i].val];for(int i = 1; i <= sz; ++i) c[i] += c[i - 1];for(int i = 1; i <= sz; ++i) a[c[t[i].val]--] = i; for(int i = sz; i; --i) Right[t[a[i]].par] += Right[a[i]];int u = root, step = 0;for(int i = 1; i <= m; ++i){int c = s2[i] - 'a';if(t[u].ch[c]) u = t[u].ch[c], ++step;else{while(u && !t[u].ch[c]) u = t[u].par;if(!u) u = root, step = 0;else{step = t[u].val + 1;u = t[u].ch[c];}}++apr[u];if(u != root) ans += (ll)(step - t[t[u].par].val) * Right[u];}for(int i = sz; i > 1; --i) f[t[a[i]].par] += f[a[i]] + apr[a[i]];for(int i = 2; i <= sz; ++i) ans += f[a[i]] * (ll)(t[a[i]].val - t[t[a[i]].par].val) * Right[a[i]];printf("%lld\n", ans);return 0; } View Code

?

轉載于:https://www.cnblogs.com/19992147orz/p/7860809.html

總結

以上是生活随笔為你收集整理的bzoj4566的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。