题意:求包含字符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;
}