BZOJ3172: [Tjoi2013]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1111  Solved: 513
[Submit][Status]

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

HINT

题解：

终于A了这题。。。

把每个单词中间用一个奇怪的字符连起来再加上辅助字符做一遍后缀数组，然后对于每个单词，找到height大于等于它的长度的向前最远和向后最远，这段长度就是它出现的次数。

实在不能再暴力了。。。

我会说直到看了题解我才知道这道题什么意思？

代码：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define inf 1000000000
#define maxn 1000000+1000
#define maxm 200
#define eps 1e-10
#define pa pair<int,int>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
int n,len,pos[maxm],ll[maxm],sa[maxn],rank[maxn],wa[maxn],wb[maxn],wr[maxn],a[maxn],h[maxn];
char s[maxn];
inline bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
        for(i=0;i<m;i++)wr[i]=0;
    for(i=0;i<n;i++)wr[x[i]=r[i]]++;
    for(i=1;i<m;i++)wr[i]+=wr[i-1];
    for(i=n-1;i>=0;i--)sa[--wr[x[i]]]=i;    
    for(j=1,p=1;p<n;j<<=1,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<m;i++)wr[i]=0;
         for(i=0;i<n;i++)wr[x[y[i]]]++;
         for(i=1;i<m;i++)wr[i]+=wr[i-1];
         for(i=n-1;i>=0;i--)sa[--wr[x[y[i]]]]=y[i];
         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
          x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
}
void geth(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;h[rank[i++]]=k)
     for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
} 
int main()
{
   freopen("input.txt","r",stdin);
   freopen("output.txt","w",stdout);    
   scanf("%d",&n);
   len = 0;
   for (int i = 0;i < n;i ++){
       scanf("%s",s + len);
       pos[i] = len;
       len = strlen(s);
       ll[i] = len - pos[i];
       s[len ++] = '$';
   }
   for (int i = 0;i < len;i ++) a[i] = s[i];
   a[len]=0;
   da(a,sa,len+1,150);
   geth(a,sa,len);
   for (int i = 0;i < n;i ++){
       int t = rank[pos[i]],j,l,r;
       for (j = t;j && h[j] >= ll[i];j --);l=j+1;
       for (j = t + 1;j <=len && h[j] >=ll[i];j ++);r=j-1;
       printf("%d\n",r-l+2);
   }
   return 0;
}