[BZOJ 3530][Sdoi 2014]数数

阿拉~好像最近总是做到 AC 自动机的题目呢喵~

题目的算法似乎马上就能猜到的样子…… AC 自动机 + 数位 dp

先暴力转移出 f[i][j] ：表示从 AC 自动机上第 j 号节点走 i 步且不碰到匹配串的方案数

然后直接用数位 dp 一位一位的试就可以了，大家都会吧~

但是…… 有前导 0 的情况真尼玛蛋疼啊！

忽的灵光一闪……

前导 0 仅能影响长度小于 L 的数的统计

那么所有长度 <L 的数全部专门暴力统计一边不就可以了！我真是特么太机智了喵~ O(∩_∩)O~

虽然有个 O(10*L*l) 的转移 但是只跑了 272 ms 呢

 #include <cstdio>
 #include <cstring>
 #define ord(ch) ((ch)-'0')
 const int sizeOfLength=;
 const int sizeOfNumber=;
 const int sizeOfType=;
 const int mod=;

 struct node
 {
     int idx;
     bool end;
     node * fail, * ch[sizeOfType];
 };
 node * dfa;
 node memory[sizeOfNumber], * port=memory;
 inline node * newnode();
 inline void insert(char * , int);
 node * q[sizeOfNumber]; int l, r;
 inline void build();
 inline void dynamicprogramming();

 char N[sizeOfLength]; int L;
 int M;
 int f[sizeOfLength][sizeOfNumber];
 char s[sizeOfNumber]; int len;
 inline int getint();
 inline int getstr(char * );
 inline void putint(int);

 int main()
 {
     int ans=;
     bool find=true;
     node * t;

     L=getstr(N);
     M=getint();
     dfa=newnode();
     for (int i=;i<=M;i++)
     {
         len=getstr(s);
         insert(s, len);
     }
     build();
     dynamicprogramming();

     t=dfa;
     for (int i=;i<L;i++)
     {
         for (int j=(i==);j<ord(N[i]);j++) if (!t->ch[j]->end)
             ans=(ans+f[L-i-][t->ch[j]->idx])%mod;
         t=t->ch[ord(N[i])];
         if (t->end)
         {
             find=false;
             break;
         }
     }
     for (int i=L-;i>=;i--)
         for (int j=;j<=;j++)
             ans=(ans+f[i][dfa->ch[j]->idx])%mod;

     putint(ans+find);

     return ;
 }
 inline int getint()
 {
     register int num=;
     register char ch;
     do ch=getchar(); while (ch<'' || ch>'');
     do num=num*+ch-'', ch=getchar(); while (ch>='' && ch<='');
     return num;
 }
 inline int getstr(char * str)
 {
     register int len=;
     register char ch;
     do ch=getchar(); while (ch<'' || ch>'');
     do str[len++]=ch, ch=getchar(); while (ch>='' && ch<='');
     return len;
 }
 inline void putint(int num)
 {
     char stack[];
     register int top=;
     for ( ;num;num/=) stack[++top]=num%+'';
     for ( ;top;top--) putchar(stack[top]);
     putchar('\n');
 }
 inline node * newnode()
 {
     node * ret=port++;
     ret->idx=port-memory-;
     ret->fail=NULL;
     memset(ret->ch, , sizeof ret->ch);
     return ret;
 }
 inline void insert(char * s, int l)
 {
     node * t=dfa;
     for (int i=;i<l;i++)
     {
         if (!t->ch[ord(s[i])]) t->ch[ord(s[i])]=newnode();
         t=t->ch[ord(s[i])];
     }
     t->end=true;
 }
 inline void build()
 {
     dfa->fail=dfa;
     for (int i=;i<sizeOfType;i++)
         if (dfa->ch[i])
         {
             dfa->ch[i]->fail=dfa;
             q[r++]=dfa->ch[i];
         }
         else
             dfa->ch[i]=dfa;
     for ( ;l<r;l++)
     {
         node * u=q[l];
         u->end|=u->fail->end;
         for (int i=;i<sizeOfType;i++)
             if (u->ch[i])
             {
                 u->ch[i]->fail=u->fail->ch[i];
                 q[r++]=u->ch[i];
             }
             else
                 u->ch[i]=u->fail->ch[i];
     }
 }
 inline void dynamicprogramming()
 {
     int tot=port-memory;
     for (int i=;i<tot;i++)
         if (!(dfa+i)->end)
             f[][(dfa+i)->idx]=;
     for (int i=;i<=L;i++)
         for (int j=;j<tot;j++) if (!(dfa+j)->end)
             for (int k=;k<sizeOfType;k++)
                 f[i][(dfa+j)->idx]=(f[i][(dfa+j)->idx]+f[i-][(dfa+j)->ch[k]->idx])%mod;
 }

机房不卡代码插入了 好评如潮