HDU2825 AC自動機+狀壓DP

題意:給你m個字符串,求長度爲n的包含至少k個給定串的字符串個數,答案對20090717取模

思路:因爲m的值很小,所以可以進行狀壓DP

dp[i][j][k]   i爲字符串當前長度,j爲當前走到的自動機的節點編號,k爲狀壓後的當前狀態

於是只需在自動機上面跑滿長度n,跑的過程中統計個數即可。

代碼如下:

#include<bits/stdc++.h>
using namespace std;
const int maxn=105;
const int mod=20090717;
struct AC
{
    int next[maxn][26],fail[maxn],ed[maxn];
    int root,L;
    int newnode()
    {
        for(int i=0;i<26;i++)
            next[L][i]=-1;
        ed[L++]=0;
        return L-1;
    }
    void init()
    {
        L=0;
        root=newnode();
    }
    void insert(char s[],int id)
    {
        int now=root;
        for(int i=0;s[i];i++)
        {
            if(next[now][s[i]-'a']==-1)
                next[now][s[i]-'a']=newnode();
            now=next[now][s[i]-'a'];
        }
        ed[now]=1<<id;
    }
    void bfs()
    {
        queue<int> q;
        fail[root]=root;
        for(int i=0;i<26;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();
            ed[now]|=ed[fail[now]];
            for(int i=0;i<26;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;
char s[12];
int dp[30][105][1025];
int cnt[1025];
int main()
{
    ios_base::sync_with_stdio(0);
    int n,m,num;
    for(int i=0;i<1024;i++)
        cnt[i]=__builtin_popcount(i);
    while(cin>>n>>m>>num)
    {
        if(!n&&!m&&!num)
            break;
        ac.init();
        for(int i=0;i<m;i++)
        {
            cin>>s;
            ac.insert(s,i);
        }
        ac.bfs();
        memset(dp,0,sizeof dp);
        dp[0][0][0]=1;
        int all=1<<m;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<ac.L;j++)
            {
                for(int k=0;k<all;k++)
                {
                    if(!dp[i][j][k]) continue;
                    for(int t=0;t<26;t++)
                    {
                        int to=ac.next[j][t];
                        dp[i+1][to][ac.ed[to]|k]+=dp[i][j][k];
                        dp[i+1][to][ac.ed[to]|k]%=mod;
                    }
                }
            }
        }
        int ans=0;
        for(int i=0;i<ac.L;i++)
        {
            for(int j=0;j<all;j++)
            {
                if(cnt[j]>=num)
                {
                    ans+=dp[n][i][j];
                    ans%=mod;
                }
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章