[bzoj3473][bzoj3277][后缀数组]字符串

3473: 字符串

Time Limit: 20 Sec Memory Limit: 256 MB
Submit: 661 Solved: 268
[Submit][Status][Discuss]
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 杯冬令营模拟赛

双倍的幸福
sol：
枚举每一个后缀，二分他贡献的长度，check的话就是找出一段hei使其>=mid,这个我们再二分一次，效率log^2，由于要在k个串中出现过，我们记录一个least[i]表示sa[i]到sa[least[i]]这一段出现过k个串。那么我们知道边界时就能O1询问。

这题加强版就是数据改大，只能sam。

#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int N=210000;
int n,m,k,Max;
int Sr[N],sa[N],rank[N],w[N],x[N],hei[N];
int end[N],belong[N],heimin[N][21],least[N],Log[N];
char sr[N];
inline int read()
{
    char c;
    int res,flag=0;
    while((c=getchar())>'9'||c<'0') if(c=='-')flag=1;
    res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=(res<<3)+(res<<1)+c-'0';
    return flag?-res:res;
}
inline void Sa()
{
    int m=Max,cnt,u,v,j;
    for(int i=1;i<=n;++i) w[x[i]=Sr[i]]++;
    for(int i=1;i<=m;++i) w[i]+=w[i-1];
    for(int i=n;i>=1;--i)
    sa[w[x[i]]--]=i;
    for(j=1;;j*=2)
    {
        cnt=0;
        for(int i=n-j+1;i<=n;++i) rank[++cnt]=i;
        for(int i=1;i<=n;++i)
        if(sa[i]>j) rank[++cnt]=sa[i]-j;
        for(int i=1;i<=m;++i) w[i]=0;
        for(int i=1;i<=n;++i) w[x[i]]++;
        for(int i=1;i<=m;++i) w[i]+=w[i-1];
        for(int i=n;i>=1;--i) sa[w[x[rank[i]]]--]=rank[i];
        m=0;
        for(int i=1;i<=n;++i)
        {
            u=sa[i];v=sa[i-1];
            if(x[u]!=x[v]||x[u+j]!=x[v+j]) ++m;
            rank[u]=m;
        }
        if(n==m) break;
        for(int i=1;i<=n;++i) x[i]=rank[i];
    }
    j=0;
    for(int i=1;i<=n;++i)
    {
        j=max(j-1,0);
        u=rank[i];
        v=sa[rank[i]-1];
        while(Sr[i+j]==Sr[v+j]) ++j;
        hei[u]=j;
    }

    for(int i=2;i<=n;++i)
    Log[i]=((i&-i)==i)?Log[i-1]+1:Log[i-1]; 
}
inline int sure(int l,int r)
{
    if(l==r) return hei[l];
    k=Log[r-l];
    return min(heimin[l][k],heimin[r-(1<<k)+1][k]);
}
inline bool check(int pos,int len)
{
    int fl,fr;
    pos=rank[pos];
    int l=1,r=pos,mid;
    while(l+1<r)
    {
        mid=l+r>>1;
        if(sure(mid,pos)>=len) r=mid;
        else l=mid; 
    }
    if(sure(l,pos)>=len) fl=l;
    else fl=r;
    if(hei[fl]>=len) --fl;

    int L=pos+1,R=n,Mid;
    while(L+1<R)
    {
        Mid=L+R>>1;
        if(sure(pos+1,Mid)>=len) L=Mid;
        else R=Mid;
    }
    if(sure(pos+1,R)>=len) fr=R;
    else fr=L;
    if(hei[fr]<len) fr=pos;
    if(least[fl]>fr) return 0;
    return 1;
}
ll ans[N];
int num[N];
int main()
{
//  freopen("string.in","r",stdin);
//  freopen("string.out","w",stdout);
    m=read();
    k=read();
    Max=150;
    int len;
    for(int i=1;i<=m;++i)
    {
        scanf("%s",sr);
        len=strlen(sr);
        for(int j=0;j<len;++j)
        {
            Sr[++n]=sr[j];
            belong[n]=i;
        }
        Sr[++n]=++Max;
        belong[n]=i;
        end[i]=n;
    }
    Sa();
    for(int i=1;i<=n;++i) heimin[i][0]=hei[i];
    for(int j=1;j<=20;++j)
    for(int i=1;i<=n;++i)
    {
        if(i+(1<<j)>n) break;
        heimin[i][j]=min(heimin[i][j-1],heimin[i+(1<<(j-1))][j-1]);
    }
    int l=1;
    int flag=0;
    for(int r=1;r<=n;++r)
    {
        num[belong[sa[r]]]++;
        if(num[belong[sa[r]]]==1) ++flag;
        if(flag>=k)
        {
            least[l]=r;
            while(l<=r)
            {
                num[belong[sa[l]]]--;
                if(!num[belong[sa[l]]]) --flag;
                ++l;
                if(flag>=k) least[l]=r;
                else break;
            }
        }
    }

    int r,mid;
    for(int i=1;i<=n;++i)
    {
        l=0;r=end[belong[i]]-i;
        while(l+1<r)
        {
            mid=l+r>>1;
            if(check(i,mid)) l=mid;
            else r=mid;
        }
        if(check(i,r)) ans[belong[i]]+=r;
        else ans[belong[i]]+=l; 
    }
    for(int i=1;i<m;++i)
    cout<<ans[i]<<' ';
    cout<<ans[m];
}