hdu 2825 Wireless Password AC自动机DP

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2825

发现只奥树上节点较少并且有最优子结构的问题就可以用ac自动机dp求解。

新发现了一点：

buildfail之后，之前等于-1的next值会赋值为与根节点相同分支的next值。

这是一个ac自动机上的状压dp，dp【i】【j】【k】代表用i个字符到达了节点j，并且当前覆盖了字符串集合k时有多少种情况。

在建立失败指针的时候，每个节点代表的集合值要或上其fail节点的集合值。即flag[i]|=flag[fail[i]]

根据ac自动机转移求解，最后找出所有覆盖集合>=k的dp[n][j][k]

最近感觉右手无力握紧会不自主颤抖，好难受

#include<bits/stdc++.h>
#define mod 20090717

using namespace std;


const int maxn=30+10;///多个模式串长度
const int N=30+10;///文章长度
const int lettersize=26;///看种类


char s[maxn];
char no[N];
long long dp[30][110][1030];

int pin(int i)
{
    int ans=0;
    while(i)
    {
        if(i&1)
            ans++;
        i>>=1;
    }
    return ans;
}

const int MAX=1e6+10;
struct Trie
{
    int next[MAX][lettersize],fail[MAX],end[MAX];
    int flag[MAX];
    int root,L;
    int newnode()
    {
        for (int i=0;i<lettersize;i++)
            next[L][i]=-1;
        end[L++]=0;
        return L-1;
    }
    void init()
    {
        L=0;
        root=newnode();
        memset(dp,0,sizeof dp);
        memset(flag,0,sizeof flag);
    }
    void insert(char *buf,int x)
    {
        int len=strlen(buf);
        int now=root;
        for (int i=0;i<len;i++)
        {
            if (next[now][buf[i]-'a']==-1)
                next[now][buf[i]-'a']=newnode();
            now=next[now][buf[i]-'a'];
        }
        end[now]++;
        flag[now]=1<<x;
    }
    void build ()
    {
        queue<int >Q;
        fail[root]=root;
        for (int i=0;i<lettersize;i++)
            if (next[root][i]==-1)
                next[root][i]=root;
            else
            {
                fail[next[root][i]]=root;
                Q.push(next[root][i]);
            }
        while (!Q.empty())
        {
            int now=Q.front();
            Q.pop();

            flag[now]=flag[now]|flag[fail[now]];

            for (int i=0;i<lettersize;i++)
                if (next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else
                {
                    fail[next[now][i]]=next[fail[now]][i];
                    Q.push(next[now][i]);
                }
        }
    }
}ac;


int main()
{
    int n,m,kk;
    while(~scanf("%d%d%d",&n,&m,&kk))
    {
//        printf("%d\n",pin(3));
        if(n==m&&m==kk&&n==0)
            break;
            ac.init();
        for(int i=0;i<m;i++)
                scanf("%s",s),ac.insert(s,i);
        ac.build();
//        for(int i=0;i<26;i++)
//            printf("next[4][%c]=%d\n",'a'+i,ac.next[5][i]);

//        for(int i=0;i<ac.L;i++)
//            printf("flag%d=%d\n",i,ac.flag[i]);
        dp[0][0][0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<ac.L;j++)
            {
                for(int k=0;k<(1<<m);k++)
                {
                    if(dp[i-1][j][k])
                    {
                        for(int l=0;l<26;l++)
                        {
                            int son=ac.next[j][l];
                            int flag=k|ac.flag[son];
//                            if(ac.end[son]==1)
//                                dp[i][0][flag]=(dp[i-1][j][k]+dp[i][0][flag])%mod;

                                dp[i][son][flag]=(dp[i-1][j][k]+dp[i][son][flag])%mod;
//
//                            printf("dp %d %d %d=%d\n",i,son,flag,dp[i][son][flag]);
                            if(i==6&&son==6 &&flag==1)
                                printf("%d %d %d %lld %lld\n",j,ac.end[son],k,dp[i-1][j][k],dp[6][6][1]);
                        }
                    }
                }
            }
        }
         long long ans=0;
//for(int i=1;i<=n;i++)
//            for(int j=0;j<ac.L;j++)
//                for(int k=0;k<(1<<m);k++)
//                    printf("dp %d %d %d =%lld\n",i,j,k,dp[i][j][k]);
            for(int j=0;j<ac.L;j++)
            {
                for(int k=0;k<(1<<m);k++)
                    if(pin(k)>=kk)
                        ans=(ans+dp[n][j][k])%mod;
            }
            printf("%lld\n",ans);
    }
    return 0;
}