[多校][HDU5343]MZL's Circle Zhou后缀自动机+动态规划

来源：互联网发布：自己的淘宝店铺怎么刷编辑：程序博客网时间：2024/06/06 03:47

对两个字符串都构建后缀自动机

对第一个字符串，求F[i]表示从rt到i节点有多少不同的路径（即求不同的子串的个数）；

对第二个字符串，求G[i]表示从节点i开始往下走能走出多少不同的路径（即求不同的子串的个数）；

这两个动态规划都可以用拓扑排序和ch[][]数组转移完成，第二个dp可能需要将ch[][]里面的每一条边反向。

枚举第一个后缀自动机里的节点u，若该节点的出边j不存在，则ans += F[u] + G[ ch[rt][j] ]

思路是对于相等的a+b和a'+b'，取在第一个字符串里字串较长的那个。

是在学校OJ上面提交的，一定要用unsigned long long才放过去

#include <iostream>#include <cstdio>#include <cstring>#include <queue>#include <vector>#define N 400050using namespace std;typedef long long LL;vector<int> e[N];char s[N];int n;struct SAM{int ch[N][27] , len[N] , link[N] , rd[N];LL F[N];int cnt=1 , las=1 , rt = 1;void add(int pos) {int x = s[pos] - 'a' + 1 , np = ++cnt , p = las;las = np; len[np] = len[p] + 1;while (p && !ch[p][x]) ch[p][x] = np , p = link[p];if (!p) link[np] = rt;else {int q = ch[p][x];if (len[q] == len[p] + 1)link[np] = q;else {int nq = ++cnt;len[nq] = len[p] + 1;memcpy(ch[nq],ch[q],sizeof(ch[q]));link[nq] = link[q];link[q] = link[np] = nq;while (p && ch[p][x] == q) ch[p][x] = nq , p = link[p];}}return ;}}A,B;int main() {scanf("%s",s+1); n = strlen(s+1);for (int i=1;i<=n;i++) A.add(i);scanf("%s",s+1); n = strlen(s+1);for (int i=1;i<=n;i++) B.add(i);A.F[1] = B.F[1] = 1LL;for (int i=1;i<=A.cnt;i++) A.F[i] = 1;for (int i=1;i<=B.cnt;i++) B.F[i] = 1;queue<int> q;for (int i=1;i<=B.cnt;i++) for (int j=1;j<=26;j++) if (B.ch[i][j]) e[ B.ch[i][j] ].push_back(i);for (int i=1;i<=B.cnt;i++) for (int j=0;j<(int)e[i].size();j++) B.rd[ e[i][j] ]++;for (int i=1;i<=B.cnt;i++) if (!B.rd[i]) q.push(i);while (!q.empty()) {int u = q.front(); q.pop();for (int i=0;i<(int)e[u].size();i++)B.F[ e[u][i] ] += B.F[u];for (int i=0;i<(int)e[u].size();i++)if ( --B.rd[ e[u][i] ] == 0) q.push(e[u][i]);}memset(A.F,0,sizeof(A.F)); A.F[1] = 1LL;for (int i=1;i<=A.cnt;i++)for (int j=1;j<=26;j++)if (A.ch[i][j]) A.rd[ A.ch[i][j] ]++;for (int i=1;i<=A.cnt;i++) if (!A.rd[i]) q.push(i);while (!q.empty()) {int u = q.front(); q.pop();for (int i=1;i<=26;i++) if (A.ch[u][i])A.F[ A.ch[u][i] ] += A.F[u];for (int i=1;i<=26;i++)if ( --A.rd[ A.ch[u][i] ] == 0) q.push(A.ch[u][i]);}unsigned long long ans = 0LL;for (int i=1;i<=A.cnt;i++) {for (int j=1;j<=26;j++) if (!A.ch[i][j]) ans += 1LL * A.F[i] * (B.F[ B.ch[1][j] ]);ans += 1LL * A.F[i];}cout << ans << endl;return 0;}

0 0