【BZOJ5337】【TJOI2018】—STR(后缀自动机+dp)

传送门

既然是问一堆串拼出一个串的子串的方案数,那显然是要在这个串的 S a m Sam Sam d p dp dp

考虑 f [ i ] [ j ] f[i][j] f[i][j]表示前 i i i个匹配串,匹配到 j j j的方案数
则可以暴力枚举自动机上的点看能不能从这个点开始匹配完一个串
可以的话就转移
复杂度 O ( S ∗ ∑ ∣ l e n a i ∣ ) O(S*\sum |len_{a_i}|) O(Slenai)

#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline int read(){
	char ch=getchar();
	int res=0,f=1;
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
	return res*f;
}
const int N=20005;
const ll mod=1e9+7;
int A[N],B[N],k,n,link[N],siz[N],len[N],tot,last;
map<int,int>nxt[N];
char s[N],ch[N];
ll f[105][N],ans;
inline void sa_extend(int c){
	int cur=++tot,p=last;
	len[cur]=len[p]+1,siz[cur]=1;
	for(;p&&!nxt[p][c];p=link[p])nxt[p][c]=cur;
	if(!p)link[cur]=1;
	else {
		int q=nxt[p][c];
		if(len[q]==len[p]+1)link[cur]=q;
		else{
			int clo=++tot;
			len[clo]=len[p]+1,link[clo]=link[q];
			nxt[clo]=nxt[q];
			for(;p&&nxt[p][c]==q;p=link[p])nxt[p][c]=clo;
			link[q]=link[cur]=clo;
		}
	}
	last=cur;
}
int main(){
	last=tot=1;
	k=read();
	scanf("%s",s);
	int ln=strlen(s);
	for(int i=0;i<ln;i++)sa_extend(s[i]-'A');
	for(int i=1;i<=tot;i++)A[len[i]]++;
	for(int i=1;i<=tot;i++)A[i]+=A[i-1];
	for(int i=1;i<=tot;i++)B[A[len[i]]--]=i;
	for(int i=tot;i>=1;i--)siz[link[B[i]]]+=siz[B[i]];
	f[0][1]=1;
	for(int i=1;i<=k;i++){
		int T=read();
		while(T--){
			scanf("%s",s);
			ln=strlen(s);
			for(int p=1;p<=tot;p++){
				if(f[i-1][p]==0)continue;
				int now=p;
				for(int j=0;j<ln;j++){
					if(!nxt[now][s[j]-'A']){
						now=-1;break;
					}
					else now=nxt[now][s[j]-'A'];
				}
				if(now!=-1)(f[i][now]+=f[i-1][p])%=mod;
			}
		}
	}
	for(int i=1;i<=tot;i++){
		(ans+=1ll*siz[i]*f[k][i])%=mod;
	}
	cout<<ans<<'\n';
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值