3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 3981 Solved: 1934
[ Submit][ Status][ Discuss]
Description
某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
Input
第一个一个整数N,表示有多少个单词,接下来N行每行一个单词。每个单词由小写字母组成,N<=200,单词长度不超过10^6
Output
输出N个整数,第i行的数字表示第i个单词在文章中出现了多少次。
Sample Input
3
a
aa
aaa
a
aa
aaa
Sample Output
6
3
1
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);
}
}