題目鏈接：點擊查看

題目大意：給出一個字符串 s 和一個字符 ch，問字符串 s 中有多少個本質不同的子串包含字符 ch

題目分析：其實就是求本質不同的子串加了一點條件，對于每個sa[ i ]，其本質不同的子串的個數原本為 len - sa[ i ] - height[ i ]，現在需要比較一下這個值與包含sa[ i ]在內，之后首次出現字符 ch 的距離所組成的后綴長度，具體實現看代碼就好了，其實用后綴自動機也可以做，因為每次添加一個字符后，會產生 len[ i ] - len[ fa[ i ] ] 個本質不同的字符串，且都是連續的，可以二分找到最長含有字符 ch 的位置，其實本質上和后綴數組的做法一樣，只不過時間復雜度可能會小一點

代碼：
?

#include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<unordered_map> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;char str[N];int sa[N]; //SA數組，表示將S的n個后綴從小到大排序后把排好序的 //的后綴的開頭位置順次放入SA中 int t1[N],t2[N],c[N];int rk[N],height[N],len;int s[N];int nx[N];void build_sa(int s[],int n,int m)//n為添加0后的總長 {int i,j,p,*x=t1,*y=t2;for(i=0;i<m;i++) c[i]=0;for(i=0;i<n;i++) c[x[i]=s[i]]++;for(i=1;i<m;i++) c[i]+=c[i-1];for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;for(j=1;j<=n;j<<=1) {p=0;for(i=n-j;i<n;i++) y[p++]=i;for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;for(i=0;i<m;i++) c[i]=0;for(i=0;i<n;i++) c[x[y[i]]]++;for(i=1;i<m;i++) c[i]+=c[i-1];for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];swap(x,y);p=1,x[sa[0]]=0;for(i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;if(p>=n) break;m=p;} }void get_height(int s[],int n)//n為添加0后的總長 {int i,j,k=0;for(i=0;i<=n;i++)rk[sa[i]]=i;for(i=0;i<n;i++) {if(k) k--;j=sa[rk[i]-1];while(s[i+k]==s[j+k]) k++;height[rk[i]]=k;} }void solve(int base=128) {build_sa(s,len+1,base);get_height(s,len); }int main() { //#ifndef ONLINE_JUDGE // freopen("input.txt","r",stdin); // freopen("output.txt","w",stdout); //#endif // ios::sync_with_stdio(false);int w;cin>>w;int kase=0;while(w--){char ch[5];scanf("%s%s",ch,str);len=strlen(str);for(int i=0;i<len;i++)s[i]=str[i]-'a'+1;s[len]=0;solve(30);//跑后綴數組 nx[len]=nx[len+1]=inf;for(int i=len-1;i>=0;i--)//nx[i]:第i位后首次出現ch字符的位置 {nx[i]=nx[i+1];if(str[i]==ch[0])nx[i]=i;}LL ans=0;for(int i=1;i<=len;i++){if(nx[sa[i]]==inf)//后面沒有符合條件的字符串了 continue;ans+=min(len-sa[i]-height[i],len-nx[sa[i]]);}printf("Case #%d: %lld\n",++kase,ans);}return 0; }

?

總結

以上是生活随笔為你收集整理的HDU - 5769 Substring(后缀数组)的全部內容，希望文章能夠幫你解決所遇到的問題。

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