Codeforces 後綴數組題目彙總

原創

2020-06-17 04:54

模板1（字符串從0開始讀入）

const int N=2e5+10,inf=0x3f3f3f3f;
int sa[N];
int rk[N];
int tmp[N];
int lcp[N];
char s[N],t[N];
int n,k;
bool cmp(int i,int j){
    if(rk[i] != rk[j]) return rk[i]<rk[j];
    else
    {
        int ri=i+k<=n?rk[i+k]:-1;
        int rj=j+k<=n?rk[j+k]:-1;
        return ri<rj;
    }
}
void build(char *s,int *sa)
{
    n=strlen(s);
    for(int i=0;i<=n;i++){
        sa[i]=i;
        rk[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k*=2){
        sort(sa,sa+n+1,cmp);
        tmp[sa[0]]=0;
        for(int i=1;i<=n;i++){
            tmp[sa[i]]=tmp[sa[i-1]]+(cmp(sa[i-1],sa[i])?1:0);
        }
        for(int i=0;i<=n;i++){
            rk[i]=tmp[i];
        }
    }
}
void LCP(char *s,int *sa,int *lcp){
    n=strlen(s);
    for(int i=0;i<=n;i++) rk[sa[i]]=i;
    int h=0;
    lcp[0]=0;
    for(int i=0;i<n;i++){
        int j=sa[rk[i]-1];
        for (h ? h-- : 0; j + h < n&&i + h < n&&s[j + h] == s[i + h]; h++);
        lcp[rk[i]-1] = h;
    }
}

模板2(字符串從1開始讀入)

n是字符串長度，m爲出現的字符種類個數，一般爲122，即‘z’的ascall碼

char s[maxn];
int y[maxn],x[maxn],c[maxn],sa[maxn],rk[maxn],height[maxn];
int n,m;
void get_SA() {
    for (int i=1; i<=n; ++i) ++c[x[i]=s[i]];
    for (int i=2; i<=m; ++i) c[i]+=c[i-1];
    for (int i=n; i>=1; --i) sa[c[x[i]]--]=i;
    for (int k=1; k<=n; k<<=1) {
        int num=0;
        for (int i=n-k+1; i<=n; ++i) y[++num]=i;
        for (int i=1; i<=n; ++i) if (sa[i]>k) y[++num]=sa[i]-k;
        for (int i=1; i<=m; ++i) c[i]=0;
        for (int i=1; i<=n; ++i) ++c[x[i]];
        for (int i=2; i<=m; ++i) c[i]+=c[i-1];
        for (int i=n; i>=1; --i) sa[c[x[y[i]]]--]=y[i],y[i]=0;
        swap(x,y);
        x[sa[1]]=1;
        num=1;
        for (int i=2; i<=n; ++i)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num;
        if (num==n) break;
        m=num;
    }
}
void get_height() {
    int k=0;
    for (int i=1; i<=n; ++i) rk[sa[i]]=i;
    for (int i=1; i<=n; ++i) {
        if (rk[i]==1) continue;//第一名height爲0
        if (k) --k;//h[i]>=h[i-1]-1;
        int j=sa[rk[i]-1];
        while (j+k<=n && i+k<=n && s[i+k]==s[j+k]) ++k;
        height[rk[i]]=k;//h[i]=height[rk[i]];
    }
}

123D. String

題意：給你一個串，求所有不同字串的貢獻和，每種字串的貢獻，k*(k+1)/2，k爲該子串出現的次數

戳題解

codeforces 204E. Little Elephant and Strings

codeforces 427D. Match & Catch

題意：求a b兩個串的最短公共子串長度，且這個子串在a串只出現過一次，在b串也只出現過一次

戳題解

codeforces 452E. Three strings

題意：給三個串s1,s2,s3,對於每個長度L（1<=L<=min(length(s1,s2,s3))）求有多少個三元組<p1,p2,p3>使得s1[p1...p1+L-1]==s2[p2..p2+L-1]==s3[p3..p3+L-1],求出所有L對應的答案對1e9+7取模。

戳題解

codeforces 802I. Fake News (hard)

題意：求一個串中，所有本質不同子串的出現次數的平方和。