HDU 5769 后缀数组

点击打开链接

题意：给一个字符，下面一个串，问你下面的串可以组成的子串中有多少个包含这个字符的

思路：比较简单的一个后缀数组，比赛的时候因为卡在了其他题上，就只看了一眼，并没有写，赛后补上感觉真的很简单，因为后缀数组可以处理重复子串所以我们直接用它的sa和height数组就行了，因为要满足有给定的字符，所以我们将sa[i]代表的串里有这个字符的全部标记起来，只有这些才会对我们的结果有贡献值，然后正常的我们算不重复子串的个数是len-sa[i]-height[i]，但是现在有这样的情况，就是字符是c，而sa[i]是dddcddd，那么它能贡献的值就是len-max((sa[i]+height[i]),c的位置)，因为若sa[i]+height[i]的值大于c的位置，这与正常的没有区别，但是若c的位置大于sa[i]+height[i]，那么这个sa能贡献的就只有c位置和之后的串，所遇这里取个最大值就行了   PS：多校总是卡题，而这些感觉简单的反而看不到(/ □ \)（弱哭）

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int MAXN=100010;
int sa[MAXN];
int t1[MAXN],t2[MAXN],c[MAXN];
int Rank[MAXN],height[MAXN];
void construct_sa(int s[],int n,int m){
    int i,j,p,*x=t1,*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(j=1;j<=n;j<<=1){
        p=0;
        for(i=n-j;i<n;i++)y[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(i=0;i<m;i++)c[i]=0;
        for(i=0;i<n;i++)c[x[y[i]]]++;
        for(i=1;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]+j]==y[sa[i]+j]?p-1:p++;
        if(p>=n)break;
        m=p;
    }
}
void construct_lcp(int s[],int n){
    int k=0;
    for(int i=0;i<=n;i++) Rank[sa[i]]=i;
    for(int 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;
    }
}
int fvis[MAXN],pr[MAXN],s[MAXN];
char str[MAXN];
int main(){
    int T,cas=1;
    char ch[10];
    scanf("%d",&T);
    while(T--){
        scanf("%s",ch);
        scanf("%s",str);
        int n=strlen(str),flag=0,fpr;
        for(int i=0;i<=n;i++) s[i]=str[i];
        memset(fvis,0,sizeof(fvis));
        memset(pr,0,sizeof(pr));
        for(int i=n-1;i>=0;i--){
            if(str[i]==ch[0]){
                flag=1;fpr=i;
            }
            if(flag) fvis[i]=1,pr[i]=fpr;
        }
        ll ans=0;
        construct_sa(s,n+1,128);
        construct_lcp(s,n);
        for(int i=0;i<=n;i++){
            if(fvis[sa[i]]) ans=ans+n-max((sa[i]+height[i]),pr[sa[i]]);
        }
        printf("Case #%d: %I64d\n",cas++,ans);
    }
    return 0;
}