2016MUTC4-1006 substring

来源：互联网发布：网络电视机顶盒wifi 编辑：程序博客网时间：2024/05/17 21:32

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

0 0