正題

題目鏈接:https://ac.nowcoder.com/acm/contest/72/F

---

題目大意

$n$個字符串，包括小寫字母和$\#$。其中$\#$可以替換為任意字符串。求有多少對字符串可能相同。

保證每個字符串至少有一個$\#$。

$2\le n\le 500000,{\sum }_{i=1}^n|s_i|\le 10^6$

---

解題思路

因為可以替換為任意字符串，所以只需要考慮第一個$\#$前和最后一個$\#$后的部分。

在仔細考慮一下，這個字符串分成前后的兩個部分$s,t$。數對$(x,y)$滿足條件當且僅當$s_x$是$s_y$的前綴，或者$s_y$是$s_x$的前綴，且$t_x$是$t_y$的后綴，或者$t_y$是$t_x$的后綴。

放到兩棵$Trie$樹上就是都有祖先關系就好了，直接跑第一棵上，然后用兩個樹狀數組在第二棵樹上維護就好了。

時間復雜度$O(m\log m)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> #include<vector> #define lowbit(x) (x&-x) using namespace std; const int N=1e6+10; int n,cnt,pos[N],dfn[N],ed[N]; long long ans;char s[N]; vector<int> G[N]; struct TreeBinary{int t[N];void Change(int x,int val){while(x<=cnt){t[x]+=val;x+=lowbit(x);}return;}int Ask(int x){int ans=0;while(x){ans+=t[x];x-=lowbit(x);}return ans;} }Bf,Bs; struct Trie{int t[N][26],m=1;int Insert(char *s,int n){int x=1;for(int i=1;i<=n;i++){if(s[i]=='#')break;int c=s[i]-'a';if(!t[x][c])t[x][c]=++m;x=t[x][c];}return x;} }Tp,Ts; void dfs(int x){dfn[x]=++cnt;for(int i=0;i<26;i++)if(Ts.t[x][i])dfs(Ts.t[x][i]);ed[x]=cnt; } void work(int x){for(int i=0;i<G[x].size();i++){int p=G[x][i];ans+=Bf.Ask(dfn[pos[p]]);Bf.Change(dfn[pos[p]],1);Bf.Change(ed[pos[p]]+1,-1);ans+=Bs.Ask(ed[pos[p]])-Bs.Ask(dfn[pos[p]]); Bs.Change(dfn[pos[p]],1);}for(int i=0;i<26;i++)if(Tp.t[x][i])work(Tp.t[x][i]);for(int i=0;i<G[x].size();i++){int p=G[x][i];Bs.Change(dfn[pos[p]],-1);Bf.Change(dfn[pos[p]],-1);Bf.Change(ed[pos[p]]+1,1);} } int main() {scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%s",s+1);int l=strlen(s+1);int x=Tp.Insert(s,l);G[x].push_back(i);reverse(s+1,s+1+l);pos[i]=Ts.Insert(s,l);}dfs(1);work(1);printf("%lld\n",ans);return 0; }

總結

以上是生活随笔為你收集整理的Wannafly挑战赛10F-小H和遗迹【Trie,树状数组】的全部內容，希望文章能夠幫你解決所遇到的問題。

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