hdu2222 keyword search

傳送門
匹配的時候不斷跳failfail,我們需要標記是否訪問。如果訪問過就說明這整條failfail鏈上的都被統計過了,所以不需要再往上跳了。節點上的計數器清不清零無所謂,因爲不會再往上走。注意初始化tottotansans

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=5e5+10;
int n,m,len,now,u,v,tot,T_T,ans;
char S[60],T[1000010];
struct node{int son[26],vis,fail,end;}a[maxn];
inline void clear(int x){
	a[x].fail=a[x].end=a[x].vis=0;
	memset(a[x].son,0,sizeof a[x].son);
}
inline void build(){
	scanf("%s",S),len=strlen(S),now=0;
	for(int i=0;i<len;++i){
		v=S[i]-'a';
		if(!a[now].son[v]) clear(++tot),a[now].son[v]=tot;
		now=a[now].son[v];
	}a[now].end++;
}
inline void getfail(){
	queue<int> Q;
	for(int i=0;i<26;++i) if(v=a[0].son[i])
		a[v].fail=0,Q.push(v);
	while(!Q.empty()){
		u=Q.front(),Q.pop();
		for(int i=0;i<26;++i){
			if(v=a[u].son[i]) a[v].fail=a[a[u].fail].son[i],Q.push(v);
			else a[u].son[i]=a[a[u].fail].son[i];
		}
	}
}
inline void Query(){
	scanf("%s",T),m=strlen(T),now=0;
	for(int i=0;i<m;++i){
		v=T[i]-'a',now=a[now].son[v];
		for(int j=now;j&&(!a[j].vis);j=a[j].fail)
			ans+=a[j].end,a[j].vis=1;
	}
}
int main(){
	scanf("%d",&T_T);
	while(T_T--){
		scanf("%d",&n),clear(0),tot=ans=0;
		for(int i=1;i<=n;++i) build();
		getfail(),Query();
		printf("%d\n",ans);
	}
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章