HDU 2825 Wireless Password AC自动机+dp

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#define REP(i,a,b) for(int i=a;i<=b;i++)
#define MS0(a) memset(a,0,sizeof(a))

using namespace std;

typedef long long ll;
const int maxn=252;
const int INF=1e9+10;

const ll MOD=20090717LL;

ll qpow(ll n,ll k,ll p)
{
    ll res=1;
    while(k){
        if(k&1) res=(res*(n%p))%p;
        n=((n%p)*(n%p))%p;
        k>>=1;
    }
    return res;
}

ll Cnt(ll k)
{
    ll res=0;
    for(ll i=0;i<=10;i++){
        if(k&(1<<i)) res++;
    }
    return res;
}

int L,M,K;
int dp[252][26][(1<<10)+10];
struct Trie
{
    int ch[maxn][26];
    int f[maxn];
    int last[maxn];
    int End[maxn];
    int rt,tot;
    int newnode()
    {
        ++tot;
        memset(ch[tot],-1,sizeof(ch[tot]));
        End[tot]=0;
        return tot;
    }
    void init()
    {
        tot=-1;
        rt=newnode();
    }
    void insert(char *s,int ID)
    {
        int len=strlen(s),u=rt;
        REP(i,0,len-1){
            int c=s[i]-‘a‘;
            if(ch[u][c]==-1) ch[u][c]=newnode();
            u=ch[u][c];
        }
        End[u]=(1<<ID);
    }
    void build()
    {
        queue<int> q;
        f[rt]=rt;last[rt]=rt;
        REP(c,0,25){
            if(~ch[rt][c]) f[ch[rt][c]]=rt,q.push(ch[rt][c]);
            else ch[rt][c]=rt;
        }
        while(!q.empty()){
            int u=q.front();q.pop();
            REP(c,0,25){
                if(~ch[u][c]) f[ch[u][c]]=ch[f[u]][c],q.push(ch[u][c]);
                else ch[u][c]=ch[f[u]][c];
            }
            if(End[f[u]]) last[u]=f[u];
            else last[u]=last[f[u]];
        }
    }
    int get(int u)
    {
        if(u==rt) return 0;
        return End[u]|get(last[u]);
    }
    int dfs(int u,int L,int k)
    {
        int &res=dp[u][L][k];
        if(~res) return res;
        if(Cnt(k)>=K) return res=qpow(26,L,MOD);
        if(L==0) return res=0;
        res=0;
        int nk=0;
        REP(c,0,25){
            int v=ch[u][c];
            nk=k;
            if(End[v]) nk|=get(v);
            else if(last[v]) nk|=get(last[v]);
            res=(res+dfs(ch[u][c],L-1,nk))%MOD;
        }
        return res;
    }
};Trie ac;
char s[maxn];

int main()
{
    while(~scanf("%d%d%d",&L,&M,&K)){
        if(L==0&&M==0&&K==0) break;
        //cout<<"L="<<L<<" M="<<M<<" K="<<K<<endl;
        ac.init();
        REP(i,1,M){
            scanf("%s",s);
            if(strlen(s)>L) continue;
            ac.insert(s,i-1);
        }
        ac.build();
        memset(dp,-1,sizeof(dp));
        printf("%I64d\n",ac.dfs(ac.rt,L,0));
    }
    return 0;
}