poj 1625 AC自动机上的DP

好题！搞了N久终于AC！

【题意】

给定N个字符集合和P个禁用的字符串，求长度为M的合法字符串个数。

【题解】

显然是AC自动机上的DP。

令dp[i][j]表示长度为i，状态为j的字符串个数。

按自动机的图转移即可，但要注意细节。

贡献几组数据。

50 50 10
qwertyuiop[]\asdfghjkl;'zxcvbnm,./ QWERTYUIOP{}|AS
aegaegu
etoijqt
tquqi
witowwt
zxcjnc
oeit
potieq
iojge
nvoq
piqper

ans=8881647922834867244791415981705771412427494861672253136057167374729235842468240763290

1 1 1
a
a

ans=0

5 10 3
abcde
abc
bc
c

ans=1048576

#include <iostream>#include <queue>using namespace std;const int maxn=800,maxm=55,base=100000000,kind=55;struct node{       int sum,id;       node* ch[kind];       node* fail;}a[maxn];struct num{       int g[20];       int len;       num()       {            memset(g,0,sizeof(g));            len=1;       }}dp[maxm][maxn];char s[maxm];int dx[400];int n,m,p,tot=0;node* getroot(){      tot++;      int i;      for (i=0;i<kind;i++)          a[tot].ch[i]=0;      a[tot].fail=0;      a[tot].id=tot;      a[tot].sum=0;      return &a[tot];}node* root=getroot();node* getnode(){      tot++;      int i;      for (i=0;i<kind;i++)          a[tot].ch[i]=0;      a[tot].fail=root;      a[tot].id=tot;      a[tot].sum=0;      return &a[tot];}void ins(char* s){     int i,k,len=strlen(s);     node* now=root;     for (i=0;i<len;i++)     {         k=dx[s[i]+128];         if (now->ch[k]==0)            now->ch[k]=getnode();         now=now->ch[k];     }     now->sum++;} void init(){     int i;     char st[maxm];     scanf("%d%d%d\n",&n,&m,&p);     gets(s);     for (i=0;i<n;i++)         dx[s[i]+128]=i;     for (i=1;i<=p;i++)     {         scanf("%s",st);         ins(st);     }}queue<node*> q;void build(){     int i,k;     node* now,*ff;     q.push(root);     while (!q.empty())     {           now=q.front();           q.pop();           for (i=0;i<kind;i++)               if (now->ch[i]!=0)               {                  q.push(now->ch[i]);                  for (ff=now->fail;ff!=0;ff=ff->fail)                      if (ff->ch[i]!=0)                      {                         now->ch[i]->fail=ff->ch[i];                         if (ff->ch[i]->sum==1) now->ch[i]->sum=1;                         break;                      }               }     }}num operator +(num a,num b){    num c=num();    int i,t,k=0,len=max(a.len,b.len);    for (i=1;i<=len;i++)    {        t=a.g[i]+b.g[i]+k;        k=t/base;        c.g[i]=t%base;    }    while (k)    {          c.g[++len]=k%base;          k/=base;    }    c.len=len;    return c;}void DP(){     int i,j,k;     dp[0][1].g[1]=1;     for (i=0;i<m;i++)         for (j=1;j<=tot;j++)         {             node* now=&a[j];             for (k=0;k<n;k++)                 if (now->ch[k]!=0)                 {                    int ct=now->ch[k]->id;                    if (now->ch[k]->sum==0) dp[i+1][ct]=dp[i+1][ct]+dp[i][j];                 }                 else                 {                     node* tmp=now->fail;                     for (;tmp!=0;tmp=tmp->fail)                         if (tmp->ch[k]!=0)                         {                            tmp=tmp->ch[k];                            break;                         }                     if (tmp==0) tmp=root;                     int ct=tmp->id;                     if (tmp->sum==0) dp[i+1][ct]=dp[i+1][ct]+dp[i][j];                 }         }}void print(){     int i,j;     num ans=num();     for (i=1;i<=tot;i++)         ans=ans+dp[m][i];     j=ans.len;     printf("%d",ans.g[j]);     for (i=j-1;i>=1;i--)     {         if (ans.g[i]<base/10) printf("0");         if (ans.g[i]<base/100) printf("0");         if (ans.g[i]<base/1000) printf("0");         if (ans.g[i]<base/10000) printf("0");         if (ans.g[i]<base/100000) printf("0");         if (ans.g[i]<base/1000000) printf("0");         if (ans.g[i]<base/10000000) printf("0");         printf("%d",ans.g[i]);     }     printf("\n");}int main(){    freopen("pin.txt","r",stdin);    freopen("pou.txt","w",stdout);    init();    build();    DP();    print();    return 0;}