HDU 2825 Wireless Password（AC自动机 + 状态压缩DP 入门）

题目链接：Click here~~

题意：

给 m 个单词构成的集合，统计所有长度为 n 的串中，包含至少 k 个单词的方案数。

解题思路：

令 dp[i][j][mask] 表示长度为 i ，跑到自动机节点 j ，已经包含单词为集合 mask 的方案数。

Note：构造AC自动机时，val 记录的是会包含的状态。

#include <queue>
#include <stdio.h>
#include <string.h>
#include <algorithm>

using namespace std;

#define CLR(a,v) memset(a,v,sizeof(a))

const int mod = 20090717;

namespace Trie
{
    const int N = 255;
    const int Size = 26;
    int top;
    struct Node{
        Node *next[Size], *f;
        //int ended_cnt;
        int ended_val;
    }node[N], *root;
    inline Node* new_node()
    {
        //node[top].ended_cnt = 0;
        node[top].ended_val = 0;
        CLR(node[top].next,NULL);
        return &node[top++];
    }
    void init()
    {
        top = 0;
        root = new_node();
    }
    void insert(char *s,int val=1)
    {
        Node *u = root;
        for(int i=0;s[i];i++)
        {
            int id = s[i] - 'a';
            if(u->next[id] == NULL)
                u->next[id] = new_node();
            u = u->next[id];
        }
        //u->ended_cnt++;
        u->ended_val = val;
    }
}

namespace ACam
{
    using namespace Trie;
    void get_fail()
    {
        queue<Node*> Q;
        for(int i=0;i<Size;i++)
        {
            Node *&ch = root->next[i];
            if(!ch)
                ch = root;
            else
            {
                ch->f = root;
                Q.push(ch);
            }
        }
        while(!Q.empty())
        {
            Node *cur = Q.front();Q.pop();
            for(int i=0;i<Size;i++)
            {
                Node *&ch = cur->next[i];
                if(!ch)
                    ch = cur->f->next[i];
                else
                {
                    ch->f = cur->f->next[i];
                    ch->ended_val |= ch->f->ended_val;
                    Q.push(ch);
                }
            }
        }
    }
    int dp[26][N][1<<10];
    int bit_cnt[1<<10];
    void pre()
    {
        bit_cnt[0] = 0;
        for(int mask=1;mask<(1<<10);mask++)
        {
            bit_cnt[mask] = bit_cnt[mask>>1] + (mask & 1);
        }
    }
    int solve(int n,int m,int k)
    {
        //memset(dp,0,sizeof(dp));		//TLE
        for(int i=0;i<=n;i++)
            for(int j=0;j<top;j++)
                for(int mask=0;mask<(1<<m);mask++)
                    dp[i][j][mask] = 0;
        dp[0][0][0] = 1;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<top;j++)
            {
                for(int mask=0;mask<(1<<m);mask++)
                {
                    if(dp[i][j][mask] == 0)
                        continue;
                    for(int kk=0;kk<Size;kk++)
                    {
                        int jj = node[j].next[kk] - node;
                        (dp[i+1][jj][mask | node[jj].ended_val] += dp[i][j][mask]) %= mod;
                    }
                }

            }
        }
        int ans = 0;
        for(int j=0;j<top;j++)
            for(int mask=0;mask<(1<<m);mask++)
                if(bit_cnt[mask] >= k)
                    (ans += dp[n][j][mask]) %= mod;
        return ans;
    }
}

char str[30];

int main()
{
    ACam::pre();
    int n,m,k;
    while(scanf("%d%d%d",&n,&m,&k),n||m||k)
    {
        ACam::init();
        for(int i=0;i<m;i++)
        {
            scanf("%s",str);
            ACam::insert(str,1<<i);
        }
        ACam::get_fail();
        int ans = ACam::solve(n,m,k);
        printf("%d\n",ans);
    }
    return 0;
}