【BZOJ】【P3172】【Tjoi2013】【单词】【题解】【SAM】

来源：互联网发布：java md5工具类运用编辑：程序博客网时间：2024/06/06 04:03

传送门：http://www.lydsy.com:808/JudgeOnline/problem.php?id=3172

换了模板……

Code:

#include<bits/stdc++.h>using namespace std;const int maxn=1e6+205;int pos[maxn],son[maxn<<1][27],pre[maxn<<1],val[maxn<<1],f[maxn<<1],sum[maxn<<1],b[maxn<<1];int last,tot,n,m;char s[maxn];int pushback(int x){val[++tot]=x;return tot;}void push(int x){int p=last;int np=pushback(val[p]+1);for(;p!=-1&&!son[p][x];p=pre[p])son[p][x]=np;if(p==-1)pre[np]=0;else{int q=son[p][x];if(val[q]==val[p]+1)pre[np]=q;else{int nq=pushback(val[p]+1);memcpy(son[nq],son[q],sizeof(son[q]));pre[nq]=pre[q];pre[q]=pre[np]=nq;for(;p!=-1&&son[p][x]==q;p=pre[p])son[p][x]=nq;}}last=np;}int main(){pre[0]=-1;scanf("%d",&n);s[0]='0';for(int i=1;i<=n;i++){char c=getchar();while(!isalpha(c))c=getchar();while(isalpha(c))s[++m]=c,c=getchar();pos[i]=m;s[++m]='z'+1;}m--;for(int i=1;i<=m;i++)push(s[i]-'a');for(int i=1;i<=tot;i++)sum[val[i]]++;for(int i=2;i<=m;i++)sum[i]+=sum[i-1];for(int i=tot;i;i--)b[sum[val[i]]--]=i;int p=0;for(int i=1;i<=m;i++)p=son[p][s[i]-'a'],f[p]=1;for(int i=tot;i;i--)f[pre[b[i]]]+=f[b[i]];int i=1;pos[0]=-1;for(int i=1;i<=n;i++){int p=0;for(int j=pos[i-1]+2;j<=pos[i];j++) p=son[p][s[j]-'a'];printf("%d\n",f[p]);}return 0;}

0 0