HDU 4416 Good Article Good sentence（后缀数组）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4416

题意：给出一个串A和多个串B。求A有多少不重复的子串使得这些子串不是B中任意一个串的子串？

思路:将A和B拼在一起，求得这个串有多少个不同的子串X，将所有的B拼接在一起，求得有多少个不同的子串Y，答案为X-Y。在求不同的子串时，设串的长度为len，首先每个后缀将会贡献出len-sa[i]+1个新的前缀，这些前缀有height[i]个与其他的重的。所以将len-sa[i]+1-height[i]叠加即可。但是，要减去含有分隔符的子串，比如两个串，aa#bbbbb，则含有分隔符的有3*6=18个串。

 

const int N=300005;



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;

    }

}











i64 cal(int *r,int L,int n,int *b,int Max)

{

    da(r,sa,L+1,Max);

    calHeight(r,sa,L);

    i64 ans=(i64)L*(L+1)/2,i;

    FOR1(i,L)  ans-=h[i];

    FOR0(i,n) L-=b[i],ans-=(i64)(b[i]+1)*L,L--;

    return ans;

}



char s[N];

int b[N],n;





 int main()

 {

     int C,num=0;

     RD(C);

     while(C--)

     {

         RD(n);

         int i,j,p=128,L=0;

         FOR0(i,n+1)

         {

             if(i) r[L++]=p++;

             RD(s);

             b[i]=strlen(s);

             FOR0(j,b[i]) r[L++]=s[j];

         }

         r[L]=0;

         i64 ans=cal(r,L,n+1,b,p+1)-cal(r+b[0]+1,L-b[0]-1,n,b+1,p+1);

         printf("Case %d: %I64d\n",++num,ans);

     }

     return 0;

 }