HDU 4622 Reincarnation(后缀自动机)

forever97 posted @ 2016年8月31日 23:35 in 字符串-后缀自动机 with tags 后缀自动机 , 714 阅读

 

【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=4622

 

【题目大意】

    给出一个长度不超过2000的字符串,有不超过10000个询问,问【L,R】子串中出现的子串数目,相同子串不可重复计数。

 

【题解】

    考虑到字符串长度只有两千,我们对每个位置往后建立2000个后缀自动机,这样子就能分别计算每个位置往后出现的字符串数目并保存,对于sam上的一个节点来说,它的匹配长度与失配位置的匹配长度只差就是他们之间的子串,所以,我们在建立sam可以同时计算出现的子串数目。

 

【代码】

#include <cstdio>
#include <cstring>
#include <algorithm> 
#include <vector>
using namespace std;
const int N=4005;
char s[N];
struct sam{
	  int p,q,np,nq,cnt,last,tot,a[N][26],l[N],f[N];
	  sam(){tot=cnt=0;last=++cnt;}
	  void init(){
	      tot=cnt=0;last=++cnt;
		  memset(a,0,sizeof(a));
	      memset(l,0,sizeof(l));
	      memset(f,0,sizeof(f));
	  }
	  int val(int c){return l[c]-l[f[c]];} 
	  void extend(int c){
		    p=last;np=last=++cnt;l[np]=l[p]+1;
		    while(!a[p][c]&&p)a[p][c]=np,p=f[p];
		    if(!p){f[np]=1;tot+=val(np);}
		    else{
			      q=a[p][c];
			      if(l[p]+1==l[q]){f[np]=q;tot+=val(np);}
			      else{
				        nq=++cnt;l[nq]=l[p]+1;
				        memcpy(a[nq],a[q],sizeof(a[q]));
				        tot-=val(p)+val(q);
				        f[nq]=f[q]; f[np]=f[q]=nq;
				        tot+=val(p)+val(q)+val(np)+val(nq);
				        while(a[p][c]==q)a[p][c]=nq,p=f[p];
			      }
		    }
	  }int ans[2005][2005];
	  void CalAns(){
	      scanf("%s",s+1);
	      int len=strlen(s+1);
	      for(int i=1;i<=len;i++){
	          init();
	          for(int j=i;j<=len;j++){
	              extend(s[j]-'a');
	              ans[i][j]=tot;
	          }
	      }
	  }
	  void solve(){
	      int Q,l,r;
	      scanf("%d",&Q); 
	      while(Q--){
	          scanf("%d%d",&l,&r);
	          printf("%d\n",ans[l][r]);
	      }
	  }
}sam;
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        sam.CalAns();
        sam.solve();
    }return 0;
}

登录 *


loading captcha image...
(输入验证码)
or Ctrl+Enter