3172: [Tjoi2013]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 3981  Solved: 1934
[Submit][Status][Discuss]

Description

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

题解：

后缀暴力，求出height，向前向后枚举height值，若height值大于串的长度则ans++

#include<cstdio>#include<cstdlib>#include<cstring>const int N=1001005;int ww[N],wx[N],wy[N],wv[N];bool cmp(int *r,int x,int x1,int ln){return r[x]==r[x1]&&r[x+ln]==r[x1+ln];}void da(int *r,int *sa,int n,int m){int *x=wx,*y=wy,*t,i,j,p;for(i=0;i<m;i++) ww[i]=0;for(i=0;i<n;i++) ww[x[i]=r[i]]++;for(i=1;i<m;i++) ww[i]+=ww[i-1];for(i=n-1;i>=0;i--) sa[--ww[x[i]]]=i;for(j=1,p=1;p<n;j*=2,m=p){for(p=0,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<n;i++) wv[i]=x[y[i]];for(i=0;i<m;i++) ww[i]=0;for(i=0;i<n;i++) ww[wv[i]]++;for(i=1;i<m;i++) ww[i]+=ww[i-1];for(i=n-1;i>=0;i--) sa[--ww[wv[i]]]=y[i];for(t=x,x=y,y=t,i=1,x[sa[0]]=0,p=1;i<n;i++)x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;}}int rank[N],height[N];void calheight(int *r,int *sa,int n){int i,j,k=0;for(i=1;i<=n;i++) rank[sa[i]]=i;for(i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);return;}int r[N],sa[N];int n,len=0,l1;char s[N];int pos[300],le[300];int main(){scanf("%d",&n);for(int i=1;i<=n;i++){pos[i]=len;scanf("%s",s);l1=strlen(s);le[i]=l1;for(int j=0;j<l1;j++)r[len++]=s[j]-'a'+1;if(i!=n)r[len++]=32;}da(r,sa,len+1,256);calheight(r,sa,len);//printf("%d\n",sa[len]);for(int i=1;i<=n;i++){    int L,R;    for(L=rank[pos[i]];height[L]>=le[i];L--);    for(R=rank[pos[i]];height[R+1]>=le[i];R++);    printf("%d\n",R-L+1);    }}