傳送門

文章目錄

• 題意：
• 思路：

題意：

一篇論文由若干單詞構成，且單詞間是隔開的，給你$n$個單詞，要求你計算每個單詞在論文中出現了多少次。

$1\le n\le 200$，所有單詞總長不超過$1e6$

思路：

考慮一種比較暴力的寫法，我們將所有串仍$ac$自動機里面，讓后遍歷每個單詞，將其視為要匹配的串，每次都跑一遍，復雜度$O(n1e6)$，還帶常數，給了$1s$顯然過不去。

考慮每個單詞在其他單詞中出現的位置，都是一些前綴的后綴！所以我們可以跑每個前綴，讓后暴跳$fail$計算答案，這個過程顯然可以用$fail$樹來優化，我們將所有前綴都打上標記，讓后做樹形$dp$向上傳遞標記，這樣到某個單詞的時候，這個值就是他在所有單詞中出現的次數。

或者可以不需要跑$fail$樹，直接利用隊列中的拓撲結構轉移也可。

// Problem: 單詞 // Contest: AcWing // URL: https://www.acwing.com/problem/content/description/1287/ // Memory Limit: 128 MB // Time Limit: 1000 ms // // Powered by CP Editor (https://cpeditor.org)//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native") //#pragma GCC optimize(2) #include<cstdio> #include<iostream> #include<string> #include<cstring> #include<map> #include<cmath> #include<cctype> #include<vector> #include<set> #include<queue> #include<algorithm> #include<sstream> #include<ctime> #include<cstdlib> #include<random> #include<cassert> #define X first #define Y second #define L (u<<1) #define R (u<<1|1) #define pb push_back #define mk make_pair #define Mid ((tr[u].l+tr[u].r)>>1) #define Len(u) (tr[u].r-tr[u].l+1) #define random(a,b) ((a)+rand()%((b)-(a)+1)) #define db puts("---") using namespace std;//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); } //void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); } //void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }typedef long long LL; typedef unsigned long long ULL; typedef pair<int,int> PII;const int N=2000010,mod=1e9+7,INF=0x3f3f3f3f; const double eps=1e-6;int n; int tr[N][26],idx,ed[N],fail[N]; int cnt[N],pos[N],ans[N],cnt1[N]; string s[N]; map<string,int>mp; vector<int>v[N];int newnode() {int u=++idx;for(int i=0;i<26;i++) tr[u][i]=0;return u; }void insert(int id,string s) {int n=s.length(),p=0;for(int i=0;i<n;i++) {int u=s[i]-'a';if(!tr[p][u]) tr[p][u]=newnode();p=tr[p][u];cnt[p]++;}ed[p]=id; }void get_fail() {queue<int>q;for(int i=0;i<26;i++) if(tr[0][i]) q.push(tr[0][i]);while(q.size()) {int u=q.front(); q.pop(); for(int i=0;i<26;i++) {int p=tr[u][i];if(p) fail[p]=tr[fail[u]][i],q.push(p);else tr[u][i]=tr[fail[u]][i];}} }void dfs(int u) {for(auto x:v[u]) dfs(x),cnt[u]+=cnt[x]; }int main() { // ios::sync_with_stdio(false); // cin.tie(0);cin>>n;for(int i=1;i<=n;i++) {cin>>s[i];if(!mp.count(s[i])) mp[s[i]]=i;insert(mp[s[i]],s[i]);}get_fail();for(int i=1;i<=idx;i++) v[fail[i]].pb(i);dfs(0);// for(int i=idx;i>=1;i--) cnt[fail[v[i]]]+=cnt[v[i]];for(int i=1;i<=idx;i++) if(ed[i]) ans[ed[i]]+=cnt[i];for(int i=1;i<=n;i++) printf("%d\n",ans[mp[s[i]]]);return 0; } /**/

總結

以上是生活随笔為你收集整理的1285. 单词 ac自动机 + fail树的全部內容，希望文章能夠幫你解決所遇到的問題。

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