正題

題目鏈接:https://www.luogu.com.cn/problem/P4770

---

題目大意

給出一個長度為$n$的字符串$S$。$q$次詢問給出一個串$T$和一個區間$[L,R]$，求$T$有多少個本質不同的子串不是$S_{L～R}$的子串。

$1\le n\le 5\times 10^5,1\le Q\le 10^5,\sum |T|\le 10^6$

---

解題思路

因為給了很多$L=1,R=n$的部分分所以應該是提示我們先從這個方面考慮。
這個部分比較簡單，考慮改為求有多少個本質不同的子串在$S_{L～R}$中出現過，因為是本質不同的子串，我們可以先建一個$S$的$SAM$和一個$T$的$SAM$。

然后把$T$串拿到$S$的$SAM$上面跑，然后每次跑出來的一個匹配長度記為$len$。對于$T$的$SAM$上的每一個節點我們記錄一個$pos$表示這個節點屬于的長度位置，然后跑到這個位置的$len$就是能夠匹配的長度了，記為$ans$。然后答案就是$max\{le{n}_i?max\{an{s}_{po{s}_i},le{n}_{f{a}_i}\},0\}$（防匹配長度超出$[le{n}_{f{a}_i},le{n}_i]$的范圍）

這樣一次的時間復雜度就是$O(|T|)$的了。

然后考慮帶區間的怎么做，我們需要保證我們在$S$的$SAM$上面跳的時候需要保證這些節點都是屬于$S_{L～R}$的自動機上的，而且要保證我們提取出來的$len$也是在那個上面的。

其實如果這個節點的$endpos$類里面有$L～R$的信息就好了，這個用線段樹合并維護一下$endpos$類的信息就可以了。

時間復雜度$O((n+\sum |T|)\log n)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const int N=1e6+10; int n,q,ql,qr,p[N],c[N],rt[N],ans[N],pos[N]; char s[N]; struct SegTree{int cnt,w[N<<5],ls[N<<5],rs[N<<5];int Change(int x,int L,int R,int pos){int p=++cnt;w[p]=max(w[x],pos);if(L==R)return p;int mid=(L+R)>>1;if(pos<=mid)ls[p]=Change(ls[x],L,mid,pos),rs[p]=rs[x];else rs[p]=Change(rs[x],mid+1,R,pos),ls[p]=ls[x];return p;}int Ask(int x,int L,int R,int l,int r){if(!x)return 0;if(L==l&&R==r)return w[x];int mid=(L+R)>>1;if(r<=mid)return Ask(ls[x],L,mid,l,r);if(l>mid)return Ask(rs[x],mid+1,R,l,r);return max(Ask(ls[x],L,mid,l,mid),Ask(rs[x],mid+1,R,mid+1,r));}int Merge(int x,int y){if(!x||!y)return x+y;int p=++cnt;w[p]=max(w[x],w[y]);ls[p]=Merge(ls[x],ls[y]);rs[p]=Merge(rs[x],rs[y]);return p;} }R; struct SAM{int cnt,last,ch[N][26],len[N],fa[N];void init(){memset(ch[1],0,sizeof(ch[1]));last=cnt=1;}int Insert(int c){int p=last,np=last=++cnt;len[np]=len[p]+1;memset(ch[np],0,sizeof(ch[np]));for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;if(!p)fa[np]=1;else{int q=ch[p][c];if(len[p]+1==len[q])fa[np]=q;else{int nq=++cnt;len[nq]=len[p]+1;memcpy(ch[nq],ch[q],sizeof(ch[nq]));fa[nq]=fa[q];fa[q]=fa[np]=nq;pos[nq]=pos[q];for(;p&&ch[p][c]==q;p=fa[p])ch[p][c]=nq;}}last=np;return np;}void Build(){for(int i=1;i<=cnt;i++)c[len[i]]++;for(int i=1;i<=n;i++)c[i]+=c[i-1];for(int i=1;i<=cnt;i++)p[c[len[i]]--]=i;for(int i=cnt;i>=1;i--){int x=p[i];rt[fa[x]]=R.Merge(rt[fa[x]],rt[x]);}return;}void next(int &x,int &l,int c){while(x){if(ch[x][c]){int maxl=R.Ask(rt[ch[x][c]],1,n,1,qr)-ql+1;if(len[fa[x]]<maxl){l=min(l+1,maxl);x=ch[x][c];return;}}x=fa[x];l=len[x];}l=0;x=1;return;}ll calc(){ll prt=0;for(int i=2;i<=cnt;i++)prt+=max(len[i]-max(ans[pos[i]],len[fa[i]]),0);return prt;} }S,T; void work(char *s){int m=strlen(s+1);T.init();for(int i=1;i<=m;i++){int x=T.Insert(s[i]-'a');pos[x]=i;}int x=1,l=0;for(int i=1;i<=m;i++){int c=s[i]-'a';S.next(x,l,c);ans[i]=l;}printf("%lld\n",T.calc()); } signed main() {scanf("%s",s+1);n=strlen(s+1);S.init();for(int i=1;i<=n;i++){int x=S.Insert(s[i]-'a');rt[x]=R.Change(rt[x],1,n,i);}S.Build();scanf("%d",&q);while(q--){scanf("%s",s+1);scanf("%d%d",&ql,&qr);work(s);}return 0; }

總結

以上是生活随笔為你收集整理的P4770-[NOI2018]你的名字【SAM,线段树合并】的全部內容，希望文章能夠幫你解決所遇到的問題。

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