CF1400F x-prime Substrings

题面传送门
真的吐血。百度上说真因数是除去\(1\)与它本身，但这道题如果除去\(1\)样例一都过不去，害的我调半天。
看到\(x\)很小，所以可以把所有序列枚举出来并插入AC自动机里面去。
然后设\(dp_{i,j}\)表示第\(i\)个数匹配到自动机上第\(j\)个节点，判断一下跳fail指针时会不会碰到序列的尾部节点然后转移即可。
时间复杂度\(O(2^{x}x^2+n\sum{|S|})\)
code:

#include<cstdio>
#include<cstring>
#include<queue>
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int t,n,m,k,x,y,z,ans[10039],flag[10039],cnt,dp[2][5039],tot=1e9,a[1039],last,nows,now;
struct ac{int son[13],fail;}f[10039];
char s[1039];
inline int check(int now){
	register int i,j,ans=0;
	for(i=1;i<=now;i++){
		for(ans=0,j=i;j<=now;j++) {
			ans+=a[j];
			if(ans!=x&&x%ans==0) return 0;
		}
	}
	return 1;
}
inline void get(int x){
	register int i;now=0;
	for(i=1;i<=x;i++) now=f[now].son[a[i]]?f[now].son[a[i]]:(f[now].son[a[i]]=++cnt);ans[now]++;
}
inline void dfs(int now,int sum){
	if(sum>x) return;
 	if(sum==x){if(check(now-1))get(now-1);return;}
 	for(int i=1;i<=9;i++)a[now]=i,dfs(now+1,sum+i),a[now]=0;
} 
queue<int> q;
inline void bfs(){
	register int i;
	for(i=1;i<=9;i++) if(f[0].son[i]) q.push(f[0].son[i]);
	while(!q.empty()){
		now=q.front();q.pop();
		for(i=1;i<=9;i++) {
			if(f[now].son[i]) f[f[now].son[i]].fail=f[f[now].fail].son[i],q.push(f[now].son[i]);
			else f[now].son[i]=f[f[now].fail].son[i];
		}
	}
}
inline void find(int x){
	if(!x||flag[x]) return;flag[x]=1;
	find(f[x].fail);ans[x]|=ans[f[x].fail];
}
int main(){
	freopen("1.in","r",stdin);
	register int i,j;
	scanf("%s",s+1);n=strlen(s+1);scanf("%d",&x);
	dfs(1,0);bfs();
	for(i=1;i<=n;i++) find(i);
	memset(dp,0x3f,sizeof(dp));dp[0][0]=0;
	for(now=0,i=0;i<n;i++){
		nows=i&1;last=nows^1;
		for(j=0;j<=cnt;j++) dp[last][j]=1e9+7;
 		for(j=0;j<=cnt;j++){
			if(dp[nows][j]>=1e9) continue;
   			now=f[j].son[s[i+1]-'0'];
			dp[last][j]=min(dp[nows][j]+1,dp[last][j]);
			if(!ans[now]) dp[last][now]=min(dp[last][now],dp[nows][j]);
		}
	}
	for(i=0;i<=cnt;i++)tot=min(tot,dp[n&1][i]);
	printf("%d\n",tot);
}