【BZOJ 3172】[TJOI 2013]単語(ACオートマチックの小さな応用)
ここをクリックして問題を見る
あなたにN N N個の単語をあげて、それぞれの単語がこのN N個の単語の中で現れる回数を求めてください.
この問題は洛谷の上の板の問題のアップグレード版であるべきだ.
L i n k Link Link
【洛谷3796】【テンプレート】ACオートマトン(補強版)の問題解詳しくはブログ【洛谷3796】【テンプレート】ACオートマトン(補強版)を参照
これはA C AC自動機の簡単な運用問題です.
L i n k Link Link
A C AC自動機詳細はブログ初学AC自動機を参照
1つの比較的に暴力的なやり方は、このN N個の単語を1本のT r i e Trie Trieを建てて、それから各単語を1つずつA C AC ACオートマチックを走って、残念ながらこのようにT L E TLE TLE TLEになります.
さらに、同じ単語がA C ACオートマチックを走る結果は同じで、なぜ一緒に走らないのかを発見することができます.これでA C ACができます.
あなたにN N N個の単語をあげて、それぞれの単語がこのN N個の単語の中で現れる回数を求めてください.
関連テーマ
この問題は洛谷の上の板の問題のアップグレード版であるべきだ.
L i n k Link Link
【洛谷3796】【テンプレート】ACオートマトン(補強版)の問題解詳しくはブログ【洛谷3796】【テンプレート】ACオートマトン(補強版)を参照
A C AC自動機
これはA C AC自動機の簡単な運用問題です.
L i n k Link Link
A C AC自動機詳細はブログ初学AC自動機を参照
問題解
1つの比較的に暴力的なやり方は、このN N個の単語を1本のT r i e Trie Trieを建てて、それから各単語を1つずつA C AC ACオートマチックを走って、残念ながらこのようにT L E TLE TLE TLEになります.
さらに、同じ単語がA C ACオートマチックを走る結果は同じで、なぜ一緒に走らないのかを発見することができます.これでA C ACができます.
コード#コード#
#include
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)
#define LL long long
#define ull unsigned long long
#define swap(x,y) (x^=y,y^=x,x^=y)
#define tc() (A==B&&(B=(A=ff)+fread(ff,1,100000,stdin),A==B)?EOF:*A++)
#define pc(ch) (pp_<100000?pp[pp_++]=ch:(fwrite(pp,1,100000,stdout),pp[(pp_=0)++]=ch))
#define N 200
#define SUM 1000000
int pp_=0;char ff[100000],*A=ff,*B=ff,pp[100000];
using namespace std;
int n,rt=1,tot=1,ans,s[N+5];
string ss[N+5];
struct Trie
{
int Son[26],Next,Cnt,Vis;
}node[SUM+5];
queue<int> q;
inline void read(int &x)
{
x=0;static char ch;
while(!isdigit(ch=tc()));
while(x=(x<<3)+(x<<1)+ch-48,isdigit(ch=tc()));
}
inline void read_string(string &x)
{
x="";static char ch;
while(isspace(ch=tc()));
while(x+=ch,!isspace(ch=tc())) if(!(~ch)) return;
}
inline void write(int x)
{
if(x>9) write(x/10);
pc(x%10+'0');
}
inline void Insert(int pos,string st)
{
register int i,nxt,x=rt,len=st.length();
for(i=0;i<len;++i)
{
if(!node[x].Son[nxt=st[i]-97]) node[x].Son[nxt]=++tot;
x=node[x].Son[nxt];
}
++node[x].Cnt,s[pos]=x;
}
inline void GetNext()
{
register int i,k;q.push(rt);
while(!q.empty())
{
k=q.front(),q.pop();
for(i=0;i<26;++i)
{
if(k^rt)
{
if(!node[k].Son[i]) node[k].Son[i]=node[node[k].Next].Son[i];
else node[node[k].Son[i]].Next=node[node[k].Next].Son[i],q.push(node[k].Son[i]);
}
else
{
if(!node[k].Son[i]) node[k].Son[i]=rt;
else node[node[k].Son[i]].Next=rt,q.push(node[k].Son[i]);
}
}
}
}
inline void AC_Automation(string st,int val)// st AC , val st
{
register int i,j,x=rt,len=st.length();
for(i=0;i<len;++i)
{
if(!(x=node[x].Son[st[i]-97])) {x=rt;continue;}
int p=x;
while(p^rt) node[p].Vis+=val,p=node[p].Next;// val
}
}
int main()
{
register int i,j;
for(read(n),i=1;i<=n;++i) read_string(ss[i]),Insert(i,ss[i]);
for(GetNext(),i=1;i<=n;++i)
if(~node[s[i]].Cnt) AC_Automation(ss[i],node[s[i]].Cnt),node[s[i]].Cnt=-1;// AC , , TLE
for(i=1;i<=n;++i) write(node[s[i]].Vis),pc('
');// Trie
return fwrite(pp,1,pp_,stdout),0;
}