HDU 5769 Substring(后缀数组)

传送门：http://acm.split.hdu.edu.cn/showproblem.php?pid=5769
思路：不考虑X字符，先考虑怎么求本质不同的串有几个，那么就是枚举一遍height数组，去掉前缀相同的就可以了。
      我们考虑X字符的话，就要保证所有求得的串都要包含这个字符。那么我们事先求一个数组，pos[i]表示当前字符（包括自身）的右边的第一个X字符的位置。然后还是枚举height数组，当前后缀没有X字符就continue。如果当前的height==0的话，说明跟前面的没有前缀相同，就是直接len-pos[sa[i]]。然后height!=0的时候，就要看这个X字符在什么位置，如果在公共前缀内的话就可以不包括这个前缀了，因为前面求过了，就是len-sa[i]-height[i]。如果在公共前缀外，说明这个没有求过，就是len-pos[sa[i]]。

#include <cstdio>#include <cstring>#include <algorithm>#include <cmath>#include <cstdlib>#include <cctype>#include <string>#include <iostream>#include <vector>#include <map>#include <set>#include <stack>#include <queue>#include <ctime>using namespace std;typedef long long ll;typedef pair<int, int> pii;#define pb push_back#define mp make_pair#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define calm (l+r)>>1const int INF = 2139062143;const int maxn=3e5+10;struct SA{    #define F(x) ((x)/3+((x)%3==1?0:tb))    #define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)    int wa[maxn],wb[maxn],wv[maxn],ws[maxn];    int rank[maxn],height[maxn];    int c0(int *r,int a,int b)    {return r[a]==r[b]&&r[a+1]==r[b+1]&&r[a+2]==r[b+2];}    int c12(int k,int *r,int a,int b){        if(k==2) return (r[a]<r[b])||(r[a]==r[b]&&c12(1,r,a+1,b+1));        else return (r[a]<r[b])||(r[a]==r[b]&&wv[a+1]<wv[b+1]);    }    void sort(int *r,int *a,int *b,int n,int m){         int i;         for(i=0;i<n;i++) wv[i]=r[a[i]];         for(i=0;i<m;i++) ws[i]=0;         for(i=0;i<n;i++) ws[wv[i]]++;         for(i=1;i<m;i++) ws[i]+=ws[i-1];         for(i=n-1;i>=0;i--) b[--ws[wv[i]]]=a[i];         return;    }    void dc3(int *r,int *sa,int n,int m){         int i,j,*rn=r+n,*san=sa+n,ta=0,tb=(n+1)/3,tbc=0,p;         r[n]=r[n+1]=0;         for(i=0;i<n;i++) if(i%3!=0) wa[tbc++]=i;         sort(r+2,wa,wb,tbc,m);         sort(r+1,wb,wa,tbc,m);         sort(r,wa,wb,tbc,m);         for(p=1,rn[F(wb[0])]=0,i=1;i<tbc;i++)         rn[F(wb[i])]=c0(r,wb[i-1],wb[i])?p-1:p++;         if(p<tbc) dc3(rn,san,tbc,p);         else for(i=0;i<tbc;i++) san[rn[i]]=i;         for(i=0;i<tbc;i++) if(san[i]<tb) wb[ta++]=san[i]*3;         if(n%3==1) wb[ta++]=n-1;         sort(r,wb,wa,ta,m);         for(i=0;i<tbc;i++) wv[wb[i]=G(san[i])]=i;         for(i=0,j=0,p=0;i<ta && j<tbc;p++)         sa[p]=c12(wb[j]%3,r,wa[i],wb[j])?wa[i++]:wb[j++];         for(;i<ta;p++) sa[p]=wa[i++];         for(;j<tbc;p++) sa[p]=wb[j++];         return;    }    void calheight(int *r,int *sa,int n){         int i,j,k=0;         for(i=1;i<=n;i++) rank[sa[i]]=i;         for(i=0;i<n;height[rank[i++]]=k)         for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);         return;    }}rst;char str[maxn];int r[maxn],sa[maxn],pos[maxn];int Case;char t[2];ll cal(int len){    ll ans=0;    if(pos[sa[1]]!=-1)ans+=len-pos[sa[1]];    for(int i=2;i<=len;i++){        if(pos[sa[i]]==-1)continue;        if(rst.height[i]==0){            ans+=len-pos[sa[i]];        }        else{            int t=pos[sa[i]];            if(t<sa[i]+rst.height[i])ans+=len-sa[i]-rst.height[i];            else ans+=len-t;        }    }    return ans;}int main(){    //freopen("D://input.txt","r",stdin);    int T;scanf("%d",&T);    while(T--){        scanf("%s",t);        scanf("%s",str);        int len=strlen(str);        for(int i=0;i<len;i++)r[i]=str[i];        r[len]=0;        rst.dc3(r,sa,len+1,128);        rst.calheight(r,sa,len);        int p=-1;        for(int i=len-1;i>=0;i--){            if(str[i]==t[0]){                p=i;            }            pos[i]=p;        }        if(p==-1)printf("Case #%d: 0\n",++Case);        else printf("Case #%d: %I64d\n",++Case,cal(len));    }    return 0;}