BZOJ 3172([Tjoi2013]单词-后缀数组第一题+RMQ)

 

3172: [Tjoi2013]单词

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 268   Solved: 145
[ 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

 

 

上一次写RMQ是什么时候？(喂，离题了）

好吧……

第一题后缀数组

不想写下去了……（快哭了TNT)

这题在BZOJ上内存很容易开过（5人组-》TLE/CE/MLE/RE/AC）

大家要是这题RE把数组开小点。别忘了[RMQ*20]数组+数组之和 //省空间

 

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<iostream>

#include<algorithm>

#include<functional>

#include<cmath>

#include<cctype>

using namespace std;

#define For(i,n) for(int i=1;i<=n;i++)

#define Rep(i,n) for(int i=0;i<n;i++)

#define Fork(i,k,n) for(int i=k;i<=n;i++)

#define ForD(i,n) for(int i=n;i;i--)

#define Forp(x) for(int p=pre[x];p;p=next[p])

#define RepD(i,n) for(int i=n;i>=0;i--)

#define MEM(a) memset(a,0,sizeof(a))

#define MEMI(a) memset(a,127,sizeof(a))

#define MEMi(a) memset(a,128,sizeof(a))

#define INF (2139062143)

#define F (1000000009)

#define MAXN (300+10)

#define MAXL (1000200+10)

#define eps (1e-9)

typedef long long ll;

char s[MAXL];

int n,pre[MAXN],tai[MAXN];

int w[MAXL],sa[MAXL],wa[MAXL*2]={0},wb[MAXL*2]={0};

// x-->上一行 y->下一行sa右值   wv-->y的左值  sa-->上次排名(求) 

bool cmp(int *a,int x,int y,int l){return (a[x]==a[y]&&a[x+l]==a[y+l]);}

void suffix_array(int n,int m)

{

   int *x=wa,*y=wb;

   For(i,m) w[i]=0;

   For(i,n) w[x[i]=s[i]]++;

   Fork(i,2,m) w[i]+=w[i-1];

   ForD(i,n) sa[w[x[i]]--]=i;

   for(int j=1,p=0;p<n;j*=2,m=p)

   {

           

      p=0;

      Fork(i,n-j+1,n) y[++p]=i;

      For(i,n) if (j<sa[i]) y[++p]=sa[i]-j;

      

      For(i,m) w[i]=0;

      For(i,n) w[x[i]]++;

      For(i,m) w[i]+=w[i-1];

      

      

      ForD(i,n) sa[w[x[ y[i] ]]--]=y[i];

      //y is release

      

      p=y[sa[1]]=1;

      Fork(i,2,n)

         y[sa[i]]=(p+=(!cmp(x,sa[i-1],sa[i],j)));



      int *t=x;x=y;y=t;

   }

}

int height[MAXL],rank[MAXL]; //height[i] 表示 sa[i]与sa[i-1]的最长公共前缀 

void make_height(char *s,int n)

{

   For(i,n) rank[sa[i]]=i;

   For(i,n)

   {

      if (rank[i]==1) continue; //求height[rank[i]] 

      int j=max(0,height[rank[i-1]]-1),k=sa[rank[i]-1];

      while (s[i+j]==s[k+j]) j++;

      height[rank[i]]=j;

   }

}

int bin[MAXN]={0},f[MAXL][24]={0};

int lcp(int l,int r)

{

   int len=r-l+1,j=(int)log2(len);

   return min(f[l][j],f[r-bin[j]+1][j]);

}

int main()

{

// freopen("bzoj3172.in","r",stdin);

   scanf("%d",&n);pre[1]=1;

   For(i,n)

   {

      scanf("%s",s+pre[i]);

      tai[i]=strlen(s+pre[i]);

      s[pre[i]+tai[i]]='#';

      pre[i+1]=pre[i]+tai[i]+1;

   }

   s[pre[n+1]]=0;

   

   suffix_array(pre[n+1]-1,200);

   

   make_height(s,pre[n+1]-1);

   

   

   int logn=int(log2((double)pre[n+1]-1)+1);

   bin[0]=1;

   For(i,logn) bin[i]=bin[i-1]<<1;

   For(i,pre[n+1]-1) f[i][0]=height[i];

   

   For(j,logn)

      For(i,pre[n+1]-1-(1<<j)+1)

         f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);

   

   For(i,n)

   {

      int tot=1;

      int j=rank[pre[i]]+1;

      {

         int l=2,r=j-1,ans=0;

         while (l<=r)

         {

            int m=l+r>>1;

            if (lcp(m,j-1)>=tai[i]) ans=m,r=m-1;

            else l=m+1;             

         }

         if (ans) ans=j-1-ans+1;

         tot+=ans;

      }

      {

         int l=j,r=pre[n+1]-1,ans=0;

         while (l<=r)

         {

            int m=l+r>>1;

            if (lcp(j,m)>=tai[i]) ans=m,l=m+1;

            else r=m-1;

         }

         if (ans) ans=ans-j+1;

         tot+=ans;

      }

      cout<<tot<<endl;

   }

// while (1);

   return 0;

}