谨以此题纪念第一次做AC自动机类问题。
大意
求比n小的能够满足不含有任意一个串的数字个数。
思路
看到比n小,可以很容易的想到要数位dp,看到和多个串匹配,可以想到要用AC自动机。
那么正解就是 在AC自动机上做数位dp。
状态是
f[i][j]
f
[
i
]
[
j
]
表示第i位和AC自动机上的第j个节点匹配的方案数(然而我并没有见过这么做的,果断的gg了)。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn=2000+5;
const ll Mod=1000000000+7;
char s[maxn],s1[maxn];
struct node
{
int son[11],fail; bool mark;
}ac[maxn*100];
int m,tot,q[maxn*100],len1;
ll ans,f[2000][2505];
void insert(char *ch)
{
int len=strlen(ch),root=0;
for(int i=0; i<len; i++)
{
int &d=ac[root].son[ch[i]-'0'];
if(!d) d=++tot;
root=d;
}
ac[root].mark=true;
}
void getfail()//¹ãËѽ¨Á¢failÖ¸Õë
{
int head=0,tail=0;
for(int i=0; i<=9; i++)
{
int d=ac[0].son[i];
if(d)
{
ac[d].fail=0; tail++; q[tail]=d;
}
}
while(head<tail)
{
head++; int now=q[head];
for(int i=0; i<=9; i++)
{
int d=ac[now].son[i];
if(!d)
{
ac[now].son[i]=ac[ac[now].fail].son[i]; continue;
}
else
{
ac[d].fail=ac[ac[now].fail].son[i]; tail++; q[tail]=d;
}
}
}
}
ll dfs(int pos,int x,bool limit)
{
if(ac[x].mark) return 0;//Èç¹ûµ±Ç°µÄµãÊÇij¸ö´®µÄÖյ㣬¼´Æ¥ÅäÉÏÁË¡£
if(pos==len1-1) return 1;//×ßÍêÒ»¸öÊý×ÖÁË,ÊýλdpµÄ±ê×¼¡£
if(!limit && f[pos][x]!=-1) return f[pos][x];//¼ÇÒ仯£¬µ«Ö»¼ÇûÉϽçµÄ¡£
ll ret=0; int up;
up=limit?s[pos+1]-'0':9;//ö¾ÙÖյ㡣 µ±Ç°Î»ÒѾѡºÃÁË£¬ÕÒÏÂһλµÄ
for(int i=0; i<=up; i++)
{
ret+=dfs(pos+1,ac[x].son[i],limit && i==up); ret%=Mod;
}
if(!limit) f[pos][x]=ret;//¼ÇÒ仯
return ret;
}
int main()
{
freopen("test.in","r",stdin);
freopen("test.out","w",stdout);
scanf("%s",s);
scanf("%d",&m);
for(int i=1; i<=m; i++)
{
scanf("%s",s1);
insert(s1);
}
getfail();
memset(f,-1,sizeof(f));
len1=strlen(s);
for(int i=0; i<len1; i++)//ÕâôÏÈö¾Ù¿ÉÒÔÈ¥³ýÇ°µ¼0µÄÓ°Ïì
{
for(int j=1; j<=(i==0?s[0]-'0':9); j++)
{
ans+=dfs(i,ac[0].son[j],i==0 && j==s[0]-'0'); ans%=Mod;
// cout<<ans<<endl;
}
}
printf("%lld\n",ans);
return 0;
}