[BZOJ3277]串

bzoj

題意

給出\(n\)個串，求每個串中有多少個子串在至少\(k\)個串中出現過。

sol

廣義字尾自動機。

先建好廣義SAM，也就是把這\(n\)個串全都插到SAM裡面去，那麼可以維護每個狀态在哪些串裡面出現過。

可以給每個狀态維護一個\(set\)，然後向上合并就是直接并起來。

這裡我寫基數排序結果鬼掉了，寫成建邊然後dfs的形式才能過。難道廣義字尾自動機不能寫基數排序？還求大佬們能教教我。

然後因為是自下而上取并，是以往上跳時集合大小一定是單調不降的。是以在對每一個串在SAM上跑比對的時候，要一路跳到\(sz[now]\ge k\)的狀态，然後計算答案。

code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
using namespace std;
const int N = 2e5+5;
int n,k,last=1,tot=1,tr[N][26],fa[N],len[N],t[N],a[N],size[N];
string s[N];set<int>S[N];
void extend(int c,int id)
{
    int v=last,u=++tot;last=u;
    len[u]=len[v]+1;S[u].insert(id);
    while (v&&!tr[v][c]) tr[v][c]=u,v=fa[v];
    if (!v) fa[u]=1;
    else{
        int x=tr[v][c];
        if (len[x]==len[v]+1) fa[u]=x;x
        else{
            int y=++tot;
            memcpy(tr[y],tr[x],sizeof(tr[y]));
            fa[y]=fa[x];fa[x]=fa[u]=y;len[y]=len[v]+1;
            while (v&&tr[v][c]==x) tr[v][c]=y,v=fa[v];
        }
    }
}
int to[N],nxt[N],head[N],cnt;
void link(int u,int v){to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;}
void dfs(int u)
{
    for (int e=head[u];e;e=nxt[e])
    {
        dfs(to[e]);
        for (set<int>::iterator it=S[to[e]].begin();it!=S[to[e]].end();++it)
            S[u].insert(*it);
    }
    size[u]=S[u].size();
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for (int i=1;i<=n;++i)
    {
        cin>>s[i];last=1;
        for (int j=0,l=s[i].length();j<l;++j) extend(s[i][j]-'a',i);
    }
    for (int i=1;i<=tot;++i) if (fa[i]) link(fa[i],i);
    dfs(1);
    if (k>n) {for (int i=1;i<=n;++i) cout<<"0 ";return 0;}
    for (int i=1;i<=n;++i)
    {
        long long ans=0;
        for (int j=0,l=s[i].length(),now=1;j<l;++j)
        {
            now=tr[now][s[i][j]-'a'];
            while (size[now]<k) now=fa[now];
            ans+=len[now];
        }
        cout<<ans<<' ';
    }
    puts("");return 0;
}                

轉載于:https://www.cnblogs.com/zhoushuyu/p/8660511.html