[BZOJ4566][Haoi2016]找相同字符 后缀自动机+dp

4566: [Haoi2016]找相同字符

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 1212  Solved: 694
[Submit][Status][Discuss]

Description

给定两个字符串，求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两

个子串中有一个位置不同。

Input

两行，两个字符串s1，s2，长度分别为n1，n2。1 <=n1, n2<= 200000，字符串中只有小写字母

Output

输出一个整数表示答案

Sample Input

aabb
bbaa

Sample Output

10

HINT

 

Source

时隔许久后重新获得了后缀自动机技能。

 #include<iostream>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 #include<cstdio>
 #define maxn 200005
 #define ll long long
 using namespace std;
 inline int read() {
     int x=,f=;char ch=getchar();
     for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-;
     for(;isdigit(ch);ch=getchar()) x=x*+ch-'';
     return x*f;
 }
 struct SUF {
     int sz[maxn*],link[maxn*],step[maxn*],son[maxn*][],v[maxn*],pos[maxn*];
     int nxt[maxn*],head[maxn*],tto[maxn*],tot;
     int cnt,last;
     SUF() {last=cnt=;memset(head,-,sizeof(head));}
     void extend(int c) {
         int p=last,np=last=++cnt;step[np]=step[p]+;sz[np]=;
         while(p&&!son[p][c]) son[p][c]=np,p=link[p];
         if(!p) link[np]=;
         else {
             int q=son[p][c];
             if(step[q]==step[p]+) link[np]=q;
             else {
                 int nq=++cnt;
                 memcpy(son[nq],son[q],sizeof(son[q]));
                 link[nq]=link[q];link[q]=link[np]=nq;
                 step[nq]=step[p]+;
                 while(son[p][c]==q&&p) son[p][c]=nq,p=link[p];
             }
         }
     }
     void pre() {
         for(int i=;i<=cnt;i++) v[step[i]]++;
         for(int i=;i<=cnt;i++) v[i]+=v[i-];
         for(int i=cnt;i>=;i--) pos[v[step[i]]--]=i;
         for(int i=cnt;i>=;i--) sz[link[pos[i]]]+=sz[pos[i]];
     }
     ll f[maxn*];
     void dfs(int x) {
         f[x]+=1ll*sz[x]*(step[x]-step[link[x]]);
         for(int i=head[x];i>=;i=nxt[i]) {
             int to=tto[i];f[to]+=f[x];dfs(to);
         }
     }
     void add(int u,int v) {
         nxt[tot]=head[u];tto[tot]=v;head[u]=tot++;
     }
 }t;
 char ch[maxn];
 int main() {
     scanf("%s",ch+);int len=strlen(ch+);
     for(int i=;i<=len;i++) t.extend(ch[i]-'a');
     t.pre();
     scanf("%s",ch+);len=strlen(ch+);
     for(int i=;i<=t.cnt;i++) if(t.link[i]) t.add(t.link[i],i);t.dfs();
     int now=,l=;ll ans=;
     for(int i=;i<=len;i++) {
         int to=ch[i]-'a';
         while(now&&!t.son[now][to]) now=t.link[now];
         if(!now) now=,l=;
         else l=min(l,t.step[now])+,now=t.son[now][to];
         ans+=t.f[t.link[now]];ans+=1ll*(l-t.step[t.link[now]])*t.sz[now];
     }
     printf("%lld\n",ans);
 }