bzoj5137:[Usaco2017 Dec]Standing Out from the Herd
阿新 • • 發佈:2019-03-27
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