題目鏈接：https://www.luogu.org/problem/P3966

題意：一篇論文是由許多個單詞組成的，給出N個單詞，查詢每個單詞在文章中出現的次數。

分析：其實還是裸的AC自動機。。。

構成文本串的時候直接在單詞后面加一個特殊符號就行，查詢函數遇到這個符號就重新從根開始，其他和上一篇一模一樣。

insert,get_fail,topu函數都沒有任何變化，因為我們插入的還是原本的單詞，唯一有變化的只是文本串，所以只有用到文本串的查詢函數變化了。

查詢函數每次重頭開始，就相當于所有模式串在按順序來的一個個模式串里面匹配了。

#include<bits/stdc++.h> using namespace std; const int maxn=1e6+300;//單詞間自行添加了符號，稍做擴大 const int inf=0x3f3f3f3f; typedef long long ll; #define meminf(a) memset(a,0x3f,sizeof(a)) #define mem0(a) memset(a,0,sizeof(a)); struct node{int fail;//失配指針failint vis[26];//子節點的位置，也就是字典樹的那26個字母int end;//如果是尾節點就記錄 int ans;//用來記錄出現次數 }AC[200100]; char s[maxn];//用來輸入模式串 char ss[maxn]; //用來輸入文本串 int cnt=0;//Trie的指針 int in[maxn];//記錄入度 int m[200100],Ans[200100]; void insert(char *s,int pos){int len=strlen(s);int now=0;//字典樹的當前指針for(int i=0;i<len;i++){//Trie樹沒有這個子節點 if(AC[now].vis[s[i]-'a']==0) AC[now].vis[s[i]-'a']=++cnt;//多組輸入，需要清除 一個個清除，之前++cnt說明需要用到這個節點了 now=AC[now].vis[s[i]-'a'];}if(AC[now].end==0) AC[now].end=pos;//標記該結點是一個單詞的結尾 ,并標記這是第幾個單詞 m[pos]=AC[now].end;//記錄當前的單詞的位置，可能是它本身，也可能是它重復單詞里第一個出現的 }void get_fail(){queue<int> que;for(int i=0;i<26;i++){//把第二層的fail指針都設為0 if(AC[0].vis[i]!=0){AC[AC[0].vis[i]].fail=0;que.push(AC[0].vis[i]);in[0]++;} }while(!que.empty()){int u=que.front();que.pop();for(int i=0;i<26;i++){if(AC[u].vis[i]!=0){//如果當前結點的子節點存在，就將子節點的fail指針指向當前結點fail指針指向的結點的對應子節點處 AC[AC[u].vis[i]].fail=AC[AC[u].fail].vis[i];in[AC[AC[u].fail].vis[i]]++;//被fail指針指向的結點的入度加1 que.push(AC[u].vis[i]);}else AC[u].vis[i]=AC[AC[u].fail].vis[i];//否則直接將這個不存在的子節點指向當前結點fail指針指向結點的對應子節點處 }} }void AC_query(char* s){int len=strlen(s);int now=0;for(int i=0;i<len;i++){if(s[i]=='T'){now=0;continue;} now=AC[now].vis[s[i]-'a'];AC[now].ans++;} }void topu(){queue<int> que;for(int i=1;i<=cnt;i++)if(in[i]==0) que.push(i);while(!que.empty()){int u=que.front();que.pop();if(AC[u].end!=0) Ans[AC[u].end]=AC[u].ans;int v=AC[u].fail;if(v!=0) in[v]--,AC[v].ans+=AC[u].ans;if(in[v]==0) que.push(v);} }int main(){int n;scanf("%d",&n);cnt=0;for(int i=1;i<=n;i++){scanf("%s",s);insert(s,i);strcat(ss,s);strcat(ss,"T");}AC[0].fail=0;//結束標志 get_fail(); //求出失配指針 AC_query(ss);topu(); for(int i=1;i<=n;i++){printf("%d\n",Ans[m[i]]);}return 0; }

?

轉載于:https://www.cnblogs.com/qingjiuling/p/11377805.html

總結

以上是生活随笔為你收集整理的AC自动机 洛谷P3966 单词的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。