題目鏈接：點擊查看

題目大意：給出 n 個字符串，再給出 m 次詢問，每次詢問需要輸出區間?[ l , r ] 內的所有字符串有多少個不同的前綴，要求算法強制在線

題目分析：統計字符串的前綴，不難想到借助字典樹實現，再將問題簡化為在線求區間 [ l , r ] 內不同數字的個數，這就是主席樹的基本應用了，兩個模板搭配起來就是這個問題的答案了

代碼：
?

#include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<cassert> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100; /*主席樹*/ struct Node {int l,r;int sum; }tree[N*40];int cnt,root[N];void change(int pos,int &k,int l,int r,int val) {tree[cnt++]=tree[k];k=cnt-1;tree[k].sum+=val;if(l==r)return;int mid=l+r>>1;if(pos<=mid)change(pos,tree[k].l,l,mid,val);elsechange(pos,tree[k].r,mid+1,r,val); }int query(int rt,int pos,int l,int r) {if(l==r)return tree[rt].sum;int mid=l+r>>1;if(pos<=mid)return tree[tree[rt].r].sum+query(tree[rt].l,pos,l,mid);elsereturn query(tree[rt].r,pos,mid+1,r); } /*主席樹*/ /*字典樹*/ string s[N];int k;int trie[N][26];//儲存每一條邊 trie[節點數][字符數]int pre[N];//判斷某一個字符串是否出現過int newnode() {k++;for(int i=0;i<26;i++)trie[k][i]=0;pre[k]=0;return k; }void insert(string& s)//插入 {int pos=0;for(int i=0;i<s.size();i++){int to=s[i]-'a';if(!trie[pos][to])trie[pos][to]=newnode();pos=trie[pos][to];} }void search(string& s,int id,int n)//查找 {int pos=0;for(int i=0;i<s.size();i++){int to=s[i]-'a';pos=trie[pos][to];if(pre[pos])change(pre[pos],root[id],1,n,-1);change(id,root[id],1,n,1);pre[pos]=id;} } /*字典樹*/ void init() {root[0]=0;tree[0].l=tree[0].r=tree[0].sum=0;cnt=1;k=-1;newnode(); }int main() { #ifndef ONLINE_JUDGE // freopen("input.txt","r",stdin); // freopen("output.txt","w",stdout); #endif // ios::sync_with_stdio(false);int n;while(scanf("%d",&n)!=EOF){init();for(int i=1;i<=n;i++){cin>>s[i];insert(s[i]);}for(int i=1;i<=n;i++){root[i]=root[i-1];search(s[i],i,n);}int m;scanf("%d",&m);int ans=0;while(m--){int l,r;scanf("%d%d",&l,&r);l=(l+ans)%n+1;r=(r+ans)%n+1;if(l>r)swap(l,r);printf("%d\n",ans=query(root[r],l,1,n));}}return 0; }

?

總結

以上是生活随笔為你收集整理的HDU - 5790 Prefix(主席树+字典树)的全部內容，希望文章能夠幫你解決所遇到的問題。

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