P2408- 不同子串个数【SA】
生活随笔
收集整理的這篇文章主要介紹了
P2408- 不同子串个数【SA】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
正題
題目鏈接:https://www.luogu.com.cn/problem/P2408
題目大意
給出一個字符串,求有多少個不同的子串。
解題思路
進行后綴排序之后,對于位置iii他有n?i+1n-i+1n?i+1個后綴,然后它和排在它前面的后綴有heightiheight_iheighti?個重復的前綴,也就是這么多個重復的子串。對于更前面的后綴的重復在剛剛那一個那里已經減去過了,所以不用統計。
所以答案就是∑i=1nn?i+1?heightsai\sum_{i=1}^nn-i+1-height_{sa_i}∑i=1n?n?i+1?heightsai??
codecodecode
#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=1e5+10; ll n,m,sa[N],x[N],y[N],c[N],h[N],rk[N],ans; char s[N]; void Qsort(){for(ll i=1;i<=m;i++)c[i]=0;for(ll i=1;i<=n;i++)c[x[i]]++;for(ll i=1;i<=m;i++)c[i]+=c[i-1];for(ll i=n;i>=1;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;return; } void SA(){m=26;for(ll i=1;i<=n;i++)x[i]=s[i]-'a'+1,y[i]=i;Qsort();for(ll w=1;w<=n;w<<=1){ll p=0; for(ll i=n-w+1;i<=n;i++)y[++p]=i;for(ll i=1;i<=n;i++)if(sa[i]>w)y[++p]=sa[i]-w;Qsort();swap(x,y);x[sa[1]]=p=1;for(ll i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+w]==y[sa[i-1]+w])?p:++p;if(p==n)break;m=p;}return; } void Get_height(){ll k=0;for(ll i=1;i<=n;i++)rk[sa[i]]=i;for(ll i=1;i<=n;i++){if(rk[i]==1)continue;if(k)k--;ll j=sa[rk[i]-1];while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])k++;h[rk[i]]=k;}return; } int main() {scanf("%lld",&n);scanf("%s",s+1);SA();Get_height();for(ll i=1;i<=n;i++)ans+=n-i+1-h[sa[i]];printf("%lld",ans); }總結
以上是生活随笔為你收集整理的P2408- 不同子串个数【SA】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: P3327-[SDOI2015]约数个数
- 下一篇: P4070-[SDOI2016]生成魔咒