1. 程式人生 > >bzoj5137:[Usaco2017 Dec]Standing Out from the Herd

bzoj5137:[Usaco2017 Dec]Standing Out from the Herd

getchar() 數組 .com ref rom n-k void const bool

傳送門

不算太難,後綴數組
將所有的字符串都連起來,就是註意減\(height\)的時候會有一部分被多減,加回來就好了
代碼:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=2e5+10;char c[maxn];
int T,n,tot=30,m=1e5+1000,sa[maxn],rk[maxn],id[maxn],iid[maxn],p[maxn];
int h[maxn],x[maxn],y[maxn],num,a[maxn],len[maxn],ans[maxn],las[maxn];
int main()
{
    read(T);
    for(rg int i=1;i<=T;i++)
    {
        scanf("%s",c+1);len[i]=strlen(c+1);
        for(rg int j=1;j<=len[i];j++)p[++n]=c[j]-'a',id[n]=i,iid[n]=j;
        p[++n]=++tot;
    }
    for(rg int i=1;i<=n;i++)a[x[i]=p[i]]++;
    for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
    for(rg int i=n;i;i--)sa[a[x[i]]--]=i;
    for(rg int k=1;k<=n;k<<=1,num=0)
    {
        for(rg int i=n-k+1;i<=n;i++)y[++num]=i;
        for(rg int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
        for(rg int i=1;i<=m;i++)a[i]=0;
        for(rg int i=1;i<=n;i++)a[x[i]]++;
        for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
        for(rg int i=n;i;i--)sa[a[x[y[i]]]--]=y[i];
        for(rg int i=1;i<=n;i++)y[i]=x[i];
        x[sa[1]]=num=1;
        for(rg int i=2;i<=n;i++)
            if(y[sa[i]]!=y[sa[i-1]]||y[sa[i-1]+k]!=y[sa[i]+k])x[sa[i]]=++num;
            else x[sa[i]]=num;
        if(num>=n)break;m=num;
    }
    for(rg int i=1;i<=n;i++)rk[sa[i]]=i;
    for(rg int i=1,j,k=0;i<=n;h[rk[i++]]=k)
        for(k=k?k-1:k,j=sa[rk[i]-1];p[j+k]==p[i+k];k++);
    h[1]=0;
    for(rg int i=1;i<=n;i++)
    {
        if(p[sa[i]]>30)break;num=min(h[i],num);
        ans[id[sa[i]]]+=len[id[sa[i]]]-iid[sa[i]]+1-h[i];
        if(id[sa[i]]!=id[sa[i-1]])ans[id[sa[i-1]]]-=h[i],ans[id[sa[i-1]]]+=num,num=h[i];
    }
    for(rg int i=1;i<=T;i++)printf("%d\n",ans[i]);
}

bzoj5137:[Usaco2017 Dec]Standing Out from the Herd