LightOJ 1428 Melody Comparison(后缀数组)
题目链接：http://www.lightoj.com/volume_showproblem.php?problem=1428
题意：若一个串t是A的子串且t不包含B，称t为A的合法子串。求A不同的合法子串的个数。
思路：首先求出B在A中出现的位置,t[i]表示从A串的i位置向后最早匹配完B的位置为t[i]，注意是匹配完B。这样求出sa和height数组后，枚举i从1到n，设x=t[sa[i]]-sa[i]，也就是sa[i]这个位置最多产生x个子串，再减去min(x,height[i])就是实际产生的子串。
int r[N],sa[N],wa[N],wb[N],wd[N],rank[N],h[N];
int cmp(int *r,int a,int b,int len)
{
return r[a]==r[b]&&r[a+len]==r[b+len];
}
void da(int *r,int *sa,int n,int m)
{
int i,j,p,*x=wa,*y=wb,*t;
FOR0(i,m) wd[i]=0;
FOR0(i,n) wd[x[i]=r[i]]++;
FOR1(i,m-1) wd[i]+=wd[i-1];
FORL0(i,n-1) sa[--wd[x[i]]]=i;
for(j=1,p=1;p<n;j<<=1,m=p)
{
p=0;
FOR(i,n-j,n-1) y[p++]=i;
FOR0(i,n) if(sa[i]>=j) y[p++]=sa[i]-j;
FOR0(i,m) wd[i]=0;
FOR0(i,n) wd[x[i]]++;
FOR1(i,m-1) wd[i]+=wd[i-1];
FORL0(i,n-1) sa[--wd[x[y[i]]]]=y[i];
t=x;x=y;y=t;p=1;x[sa[0]]=0;
FOR1(i,n-1) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
}
}
void calHeight(int *r,int *sa,int n)
{
int i,j,k=0;
FOR1(i,n) rank[sa[i]]=i;
FOR0(i,n)
{
if(k) k--;
j=sa[rank[i]-1];
while(i+k<n&&j+k<n&&r[i+k]==r[j+k]) k++;
h[rank[i]]=k;
}
}
char s[N],s1[N];
int b[N],n,Next[N],pos[N],t[N];
void getNext(char s[],int len,int next[])
{
next[0]=-1;
int i=0,j=-1;
while(i<len)
{
if(j==-1||s[i]==s[j]) next[++i]=++j;
else j=next[j];
}
}
void KMP(char s1[],int n,char s2[],int m,int next[])
{
clr(pos,0);
int i=0,j=0;
while(i<n)
{
if(j==-1||s1[i]==s2[j]) i++,j++;
else j=next[j];
if(j==m) j=next[j],pos[i-1]=1;
}
clr(t,-1);
t[n]=n;
FORL0(i,n-1) if(pos[i]) t[i-m+1]=i;
FORL0(i,n-1) if(t[i]==-1) t[i]=t[i+1];
}
int deal()
{
int ans=0,i,x;
FOR1(i,n)
{
x=t[sa[i]]-sa[i];
ans+=x-min(x,h[i]);
}
return ans;
}
int main()
{
int C,num=0;
RD(C);
while(C--)
{
RD(s);
RD(s1);
getNext(s1,strlen(s1),Next);
KMP(s,strlen(s),s1,strlen(s1),Next);
int i;
n=strlen(s);
FOR0(i,n) r[i]=s[i];
r[n]=0;
da(r,sa,n+1,130);
calHeight(r,sa,n);
printf("Case %d: %d\n",++num,deal());
}
return 0;
}