【BZOJ3277】串(後綴自動機)
阿新 • • 發佈:2018-02-21
lib 集合 合並 set 構建 clas long while include
就跳父親
那麽,當前位置產生的貢獻就是\(longest\)
累加就好了
【BZOJ3277】串(後綴自動機)
題面
BZOJ
題解
廣義後綴自動機???
照著別人的打了一遍。。
相當於每個串都構建一個後綴自動機
構建完一個串之後,直接把當前的last指回root就行了???
好吧,我覺得我看了一個假的廣義後綴自動機。。
不過這道題這樣就行了
考慮如何計算一個串在多少個母串中出現過?
考慮他的\(right\)集合中有多少個母串的結束位置就好了
那麽,每個節點開個\(set\)存一下,然後按照拓撲序向上合並一次
因為\(parent\)包含的母串個數一定較大,
所以,對於每一個串計算答案
就把他直接丟到廣義後綴自動機裏面去
如果當前位置的\(size\),也就是包含的母串個數\(<K\)
就跳父親
那麽,當前位置產生的貢獻就是\(longest\)
累加就好了
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 220000
struct Line{int v,next;}e[MAX<<1];
int h[MAX<<1],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
struct Node
{
int son[26];
int ff,len;
set<int> d;
}t[MAX<<1];
int size[MAX<<1];
int last,tot;
void init(){last=tot=1 ;}
void extend(int c,int id)
{
int p=last,np=++tot;last=np;
t[np].len=t[p].len+1;
t[np].d.insert(id);
while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].ff;
if(!p)t[np].ff=1;
else
{
int q=t[p].son[c];
if(t[q].len==t[p].len+1)t[np].ff=q;
else
{
int nq=++tot;
t[nq]=t[q];
t[nq].len=t[p].len+1;
t[q].ff=t[np].ff=nq;
while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].ff;
}
}
}
void DFS(int u)
{
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;
DFS(v);
for(set<int>::iterator it=t[v].d.begin();it!=t[v].d.end();++it)
t[u].d.insert(*it);
}
size[u]=t[u].d.size();
}
int n,K;
string g[MAX];
int main()
{
ios::sync_with_stdio(false);
init();
cin>>n>>K;
for(int i=1;i<=n;++i)
{
cin>>g[i];
last=1;
for(int j=0,l=g[i].length();j<l;++j)extend(g[i][j]-97,i);
}
for(int i=1;i<=tot;++i)if(t[i].ff)Add(t[i].ff,i);
DFS(1);
if(K>n){for(int i=1;i<=n;++i)putchar('0'),putchar(' ');return 0;}
for(int i=1;i<=n;++i)
{
ll ans=0;
int now=1,len=g[i].length();
for(int j=0;j<len;++j)
{
now=t[now].son[g[i][j]-97];
while(size[now]<K)now=t[now].ff;
ans+=t[now].len;
}
printf("%lld ",ans);
}
return 0;
}
【BZOJ3277】串(後綴自動機)