HDU - 5769 Substring(後綴數組)

題目鏈接：點擊查看

題目大意：給出一個字符串 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;
}