題目：https://www.luogu.com.cn/problem/P3966

題意：給出$n$個單詞組成一篇文章，問：每個單詞分別在文章中出現(xiàn)過幾次

思路：
學習的時候參考了大佬的題解：https://www.luogu.com.cn/blog/scp-d14134/ti-xie-p3966-tjoi2013-dan-ci-post

• 這道題很容易想到ac自動機，建樹的過程中統(tǒng)計每個子串出現(xiàn)的個數(shù)，然后查詢n次，但顯然，這樣就 會$TLE$！
• 這個時候我們就需要用到一個$fail$ 樹，我們都知道ac自動機中的$fail$指針，$fail$樹就是將$fail$指針所指的值和表示的邊建出來的樹
• 這道題只需要跑一遍ac自動機，統(tǒng)計每個子串出現(xiàn)的個數(shù)。然后一個節(jié)點表示的字符串在整個文章中出現(xiàn)的次數(shù)相當于它在$fail$樹中子樹的權(quán)值的和。

代碼：

#include<bits/stdc++.h> #define INF 0x3f3f3f3f #define lowbit(x) x&(-x) #define pb push_back #define MAXN 1100010using namespace std;typedef long long ll; typedef pair<int,int>P; int trie[MAXN][30],cntword[MAXN],vis[MAXN],fail[MAXN];//trie樹，節(jié)點個數(shù)，單詞結(jié)尾點的下標,fail數(shù)組 int que[MAXN]; char s[MAXN]; int tot; void Insert(char *a,int idx) {int root=0;for(int i=0;a[i];i++){int x=a[i]-'a';if(!trie[root][x])trie[root][x]=++tot;root=trie[root][x];cntword[root]++;}vis[idx]=root; } void getfail() {int head=0,tail=0;for(int i=0;i<26;i++)if(trie[0][i])que[++tail]=trie[0][i];while(head<tail){int now=que[++head];for(int i=0;i<26;i++){if(trie[now][i]){que[++tail]=trie[now][i];fail[trie[now][i]]=trie[fail[now]][i];}else trie[now][i]=trie[fail[now]][i];}}for(int i=tot;i>=0;i--)cntword[fail[que[i]]]+=cntword[que[i]];//按照dfs序?qū)鹤庸?jié)點的值更新到父節(jié)點中 } int main() {int n;scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%s",s);Insert(s,i);}getfail();for(int i=1;i<=n;i++)printf("%d\n",cntword[vis[i]]);return 0; }

總結(jié)

以上是生活随笔為你收集整理的【AC自动机】P3966 [TJOI2013] 单词的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。