2016MUTC4-1006 substring

题意：求包含字符x的不同子串个数

题解：考虑不包含字符X的情况，不同的子串可以用后缀数组解决：

考虑含字符X的情况：即对于起点为sa[i]，终点必须在X之后：

nxt[i]表示位置i及之后的第一个X出现的位置

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

const int N=1e5+10;

typedef long long LL;
int nxt[N];
char s[N];
int sa[N],t[N],t2[N],c[N],n;
void build_sa(int m){
    int i,*x=t,*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 (int k=1;k<=n;k<<=1){
        int p=0;
        for (i=n-k;i<n;i++)y[p++]=i;
        for (i=0;i<n;i++)if (sa[i]>=k)y[p++]=sa[i]-k;
        for (i=0;i<m;i++)c[i]=0;
        for (i=0;i<n;i++)c[x[y[i]]]++;
        for (i=0;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]+k]==y[sa[i]+k]?p-1:p++;
        if (p>=n)break;
        m=p;
    }
}
int Rank[N],height[N];
void getHeight(){
    int i,j,k=0;
    for (i=0;i<n;i++)Rank[sa[i]]=i;
    for (i=0;i<n;i++){
        if (k)k--;
        int j=sa[Rank[i]-1];
        while (s[i+k]==s[j+k])k++;
        height[Rank[i]]=k;
    }
}
void work()
{
    scanf("%s",c);scanf("%s",s);
    n=strlen(s);
    if (s[n-1]==c[0])nxt[n-1]=n-1;else nxt[n-1]=n;
    for (int i=n-2;i>=0;i--)if (s[i]==c[0])nxt[i]=i;else nxt[i]=nxt[i+1];
    s[n]='#';s[n+1]='\0';n++;
    build_sa(128);
    getHeight();
    LL Ans=0;
    for (int i=1;i<n;i++)Ans+=(n-1)-max(nxt[sa[i]],sa[i]+height[i]);
    cout<<Ans<<endl;
}
int main()
{
    int Case;scanf("%d",&Case);
    for (int C=1;C<=Case;C++){
        printf("Case #%d: ",C);
        work();
    }
    return 0;
}