bzoj3473

http://www.elijahqi.win/archives/607
Description

给定n个字符串，询问每个字符串有多少子串（不包括空串）是所有n个字符串中至少k个字符串的子串？
Input

第一行两个整数n，k。
接下来n行每行一个字符串。
Output

一行n个整数，第i个整数表示第i个字符串的答案。
Sample Input

3 1
abc
a
ab
Sample Output

6 1 3
HINT

对于 100% 的数据，1<=n，k<=10^5，所有字符串总长不超过10^5，字符串只包含小写字母。

Source

Adera 1 杯冬令营模拟赛

首先这个题目我确实是读了很久才读懂的x

分别询问这n个字符串每一个字符串的子串至少在其他k个字符串中出现 的个数

由于求每个的贡献，所以不妨我们枚举每个后缀然后在每个后缀里再枚举所有前缀，这样就可以不重不漏的枚举所有子串了

为了款苏查找判断这个区间内是否有k个及以上的不同串。我们预处理一个L数组，表示L[i]…i区间内有k个及以上的不同串且L[i]最大。（也就是使以i为右端点的区间满足条件的最小区间）显然这是可以一遍O(n)扫描完成的。(用一个指针p表示左端点的移动)。则只需看[l…r]区间是否包含了[L[r]…r]区间即可快速判断。 这个区间指的是sa的区间就是按照顺序排好的后缀

我们记录每个串在总串中从data[i].st开始data[i].ed结束

假设我们在求a[j…ed]这个后缀的贡献。则他的贡献就是有几个前缀出现在k个及以上不同串中，显然如果a[j..j+len-1]满足且len最大，则对答案的贡献为len。我们只需求出这个最大的len。考虑到如果a[j-1…ed]对答案的贡献为len，则a[j…ed]对答案的贡献至少为len-1，<—(由此)我们可以像求height数组那样来求这个贡献。

还需要实现的是如何判断a[j…j+len-1]（以下记作串t）是否出现在k个及以上不同的串中。首先根据h数组我们可以求出最大的区间[l..r]满足这r-l+1个后缀均包含串t。我们统计这个区间内是否有k个及以上的不同串。

#include<cstdio>
#include<cstring>
#define N 220000
int m,nn,n1,k,n,kk;
int a[N],rank[N<<1],rank1[N],sa[N],tmp[N],count[N],height[N],fmin[N][20];
int bl[N],cnt[N],L[N],Log[N];
char str1[N];
struct node{
    int st,ed;
}data[N>>1];
inline int min(int x,int y){return x<y?x:y; }
inline int lcp(int x,int y){
    x++;int t=y-x+1;t=Log[t];
    return min(fmin[x][t],fmin[y-(1<<t)+1][t]);
}
bool check(int st,int len){ //串sa[st]->sa[st+len-1]是否满足题目的条件 
    int l1,r1;//二分的目标是找到最大的sa区间，使包含sa[st]..sa[st]+len-1 
    if (height[st]<len) l1=st;else{
        int l=1,r=st-1;
        while (l<=r){
            int mid=(l+r)>>1;
            if(lcp(mid,st)>=len) r=mid-1;else l=mid+1;
        }
        l1=l;
    }
    if (height[st+1]<len) r1=st;else{
        int l=st+1,r=n;
        while (l<=r){
            int mid=(l+r)>>1;
            if (lcp(st,mid)>=len) l=mid+1;else r=mid-1;
        }
        r1=r;
    } 
    return L[r1]>=l1;//L[r1]到r1的区间肯定包含了kk及以上不同的字串，判断l1->r1的区间能否包含 
}
int main(){
    freopen("bzoj3473.in","r",stdin);
    scanf("%d%d",&n1,&kk);
    m=30;n=1;
    for (int i=1;i<=n1;++i){
        scanf("%s",str1);nn=strlen(str1);data[i].st=n;
        for (int j=0;j<nn;++j) a[j+n]=str1[j]-'a'+1,bl[j+n]=i;
        n+=nn;data[i].ed=n-1;a[n++]=m++;
    }n-=1;
//  for (int i=1;i<=n1;++i) printf("%d %d\n",data[i].st,data[i].ed);
//  for (int i=1;i<=n;++i) printf("%d ",a[i]);  
    for (int i=1;i<=n;++i) count[a[i]]=1;
    for (int i=1;i<=n+255;++i) count[i]+=count[i-1];
    for (int i=1;i<=n;++i) rank[i]=count[a[i]];
    for (int p=1;k!=n;p<<=1,m=k){
        for (int i=1;i<=m;++i) count[i]=0;
        for (int i=1;i<=n;++i) count[rank[i+p]]++;
        for (int i=1;i<=m;++i) count[i]+=count[i-1];
        for (int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
        for (int i=1;i<=m;++i) count[i]=0;
        for (int i=1;i<=n;++i) count[rank[i]]++;
        for (int i=1;i<=m;++i) count[i]+=count[i-1];
        for (int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
        memcpy(rank1,rank,sizeof(rank)>>1);
        rank[sa[1]]=k=1;
        for (int i=2;i<=n;++i){
            if (rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
            rank[sa[i]]=k;
        }
    }
/*  for (int i=1;i<=n;++i){
        for (int j=sa[i];j<=n;++j) printf("%c",a[j]+'a'-1);printf("\n");
    }*/
    //for (int i=1;i<=n;++i) printf("%d ",rank[i]);printf("\n");
    k=0;
    for (int i=1;i<=n;++i){
        if (rank[i]==1) continue;
        k=k==0?0:k-1;
        while (a[i+k]==a[sa[rank[i]-1]+k]) ++k;
        height[rank[i]]=k;
    }
    //for (int i=1;i<=n;++i) printf("%d ",height[i]);
    Log[0]=-1;
    for (int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
//  for (int i=1;i<=n;++i) printf("%d ",Log[i]);
    for (int i=1;i<=n;++i) fmin[i][0]=height[i];
    for (int j=1;j<=Log[n];++j){
        for (int i=1;i<=n-(1<<j)+1;++i){
            fmin[i][j]=min(fmin[i][j-1],fmin[i+(1<<(j-1))][j-1]);
        }
    }
    //o(n)求l数组
    int num=0;
    for (int i=1,p=1;i<=n;++i){
        if (!bl[sa[i]])break;//因为前面的设定，如果遇到分隔符，自动退出 
        if (cnt[bl[sa[i]]]==0) num++;//如果不存在这一块 则计数器+1
        cnt[bl[sa[i]]]++;
        while (num>kk||num==kk&&cnt[bl[sa[p]]]>1){
            --cnt[bl[sa[p]]];if(cnt[bl[sa[p++]]]==0) num--;
        } 
        if(num>=kk) L[i]=p;else L[i]=-1;
    } 
    //for (int i=1;i<=n;++i) printf("%d ",L[i]);
    for (int i=1;i<=n1;++i){
        long long ans=0;int k=0;
        for (int j=data[i].st;j<=data[i].ed;++j){
            k=k==0?0:k-1;
            while (j+k<=data[i].ed&&check(rank[j],k+1)) ++k;//检查一下rank[j]这个后缀是否满足k+1这个长度的条件 
            ans+=k;
        }
        printf("%lld",ans);if (i!=n1) printf(" ");
    }
    return 0;
}