bzoj3172: [Tjoi2013]单词

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

题解：

先将所有字符串拼起来，然后求一遍后缀数组，找到第以i个串开头的后缀的位置，求出向左走保证height[x]>=len[i]的最小的x,以及向右走height[y]>=len[i]的最大的y,那么答案就是y-x+2;

向左扩展或向右扩展在扩展的过程中，min(height[x])肯定单调不增，所以可以二分答案，rmq预处理区间最小值，时间复杂度O(nlogn)

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1000000+210;
char A[maxn];
int sa[maxn],rk[maxn],tmp[maxn],c[maxn];
int n;
int minn[maxn][22];
int sum[maxn];
int len[maxn];
int height[maxn];
int lg[maxn];
int tot=0;
char B[maxn];
inline void get_sa(){
    int *rnk=rk,*tp=tmp;
    n=strlen(A+1);
    int m=300;
    for(int i=1;i<=n;i++)
        tp[i]=A[i],rnk[i]=A[i];
    for(int i=0;i<=m;i++)
        c[i]=0;
    for(int i=1;i<=n;i++)
        c[tp[i]]++;
    for(int i=1;i<=m;i++)
        c[i]+=c[i-1];
    for(int i=n;i>=1;i--)
        sa[c[tp[i]]--]=i;
    for(int j=1;j<=n;j<<=1){
        int p=0;
        for(int i=n-j+1;i<=n;i++)
            tp[++p]=i;
        for(int i=1;i<=n;i++)
            if(sa[i]>j)
                tp[++p]=sa[i]-j;
        for(int i=0;i<=m;i++)
            c[i]=0;
        for(int i=1;i<=n;i++)
            c[rnk[tp[i]]]++;
        for(int i=1;i<=m;i++)
            c[i]+=c[i-1];
        for(int i=n;i>=1;i--)
            sa[c[rnk[tp[i]]]--]=tp[i];
        swap(rnk,tp);
        rnk[sa[1]]=1;
        p=1;
        for(int i=2;i<=n;i++){
            int O1=sa[i]+j>n?-1:tp[sa[i]+j];
            int O2=sa[i-1]+j>n?-1:tp[sa[i-1]+j];
            rnk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&O1==O2)?p:++p;
        }
        m=p;
        if(m>=n)
            break;
    }
}
inline void get_height(){
    for(int i=1;i<=tot;i++)
        rk[sa[i]]=i;
    int h=0;
    for(int i=1;i<=tot;i++){
        --h=h<0?0:h;
        int u=sa[rk[i]-1];
        while(A[u+h]==A[i+h])
            h++;
        height[rk[i]]=h;
    }
}
inline void rmq_pre(){
    for(int i=1;i<=tot;i++){
        int u=1,cnt=0;
        while(u<=i)
            u<<=1,cnt++;
        lg[i]=cnt-1;
    }
    for(int i=1;i<=tot;i++)
        minn[i][0]=height[i];
    for(int j=1;j<=30;j++)
        for(int i=1;i<=tot;i++){
            if(i+(1<<j)-1>tot)
                break;
            minn[i][j]=min(minn[i][j-1],minn[i+(1<<(j-1))][j-1]);
        }
}
inline int get_min(int l,int r){
    if(l>r)
        return 0x3f3f3f3f;
    int u=lg[r-l+1];
    return min(minn[l][u],minn[r-(1<<u)+1][u]);
}
inline int get_l(int x,int length){
    int l=1,r=x;
    while(l+1<r){
        int mid=(l+r)>>1;
        int u=get_min(mid+1,x);
        if(u>=length)
            r=mid;
        else l=mid;
    }
    if(get_min(l+1,x)>=length)
        return l;
    return r;
}
inline int get_r(int x,int length){
    int l=x,r=tot;
    while(l+1<r){
        int mid=(l+r)>>1;
        int u=get_min(x+1,mid);
        if(u>=length)
            l=mid;
        else r=mid;
    }
    if(get_min(x+1,r)>=length)
        return r;
    return l;
}
inline void work(int i){
    int t=rk[sum[i-1]+1];
    int l=get_l(t,len[i]),r=get_r(t,len[i]);
    //printf("%d %d\n",l,r);
    printf("%d\n",r-l+1);
}
int main(){
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",B+1);
        len[i]=strlen(B+1);
        for(int j=1;j<=len[i];j++)
            A[++tot]=B[j];
        if(i!=n)
            A[++tot]='z'+1;
        sum[i]=sum[i-1]+len[i]+(i!=n);
    }
    get_sa();
    get_height();
    rmq_pre();
    for(int i=1;i<=n;i++)
        work(i);
    //for(int i=1;i<=tot;i++)
    //  printf("%d ",sa[i]);
return 0;
}