考研路茫茫——单词情结 HDU - 2243 （ac自动机）

考研路茫茫——单词情结

 HDU - 2243 

题意：

和上一题基本差不多

比较坑的是，矩阵快速幂的时候m+1了，会爆int，所以要用long long

#include <bits/stdc++.h>
using namespace std;
#define ull unsigned long long
#define CLR(m,a) memset(m,a,sizeof(m))
const int maxnode=35;
const int sigma=26;

struct Matrix
{
    ull n,m[maxnode<<1][maxnode<<1];
    void init(ull sz)
    {
        n=sz;
        CLR(m,0);
    }
    Matrix(ull sz){init(sz);}
    void setE()
    {
        for(int i=0;i<n;i++) m[i][i]=1;
    }
    Matrix operator *(const Matrix &a)
    {
        Matrix ans(n);
        for(int k=0;k<n;k++)
            for(int i=0;i<n;i++)
                for(int j=0;j<n;j++)
                    ans.m[i][j]=ans.m[i][j]+m[i][k]*a.m[k][j];
        return ans;
    }
};

struct AC
{
    int ch[maxnode][sigma],f[maxnode];
    int match[maxnode];
    int sz;

    void init()
    {
        CLR(ch[0],0);
        CLR(match,0);
        sz=1;
    }
    int idx(char c){return c-'a';}
    void inser(char *s)
    {
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                CLR(ch[sz],0);
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        match[u]=1;
    }
    void getfail()
    {
        queue<int> q;
        f[0]=0;
        for(int c=0;c<sigma;c++){
            int u=ch[0][c];
            if(u){
                q.push(u);
                f[u]=0;
            }
        }
        while(!q.empty()){
            int r=q.front();
            q.pop();
            for(int c=0;c<sigma;c++){
                int u=ch[r][c];
                if(!u){ch[r][c]=ch[f[r]][c];continue;}
                q.push(u);
                int v=f[r];
                while(v&&!ch[v][c]) v=f[v];
                f[u]=ch[v][c];
                match[u]|=match[f[u]];
            }
        }
    }
    ull solve(ull m)
    {
        int tot=sz*2;
        Matrix a(tot),b(tot);
        for(int i=0;i<sz;i++)
        for(int j=0;j<sigma;j++){
            if(match[i]||match[ch[i][j]]) continue;
            a.m[i][ch[i][j]]++;
        }
        for(int i=0;i<sz;i++) a.m[i][i+sz]=1;  //构造矩阵这里wa了几次。。。
        for(int i=sz;i<tot;i++) a.m[i][i]=1;
        b.setE();
        ull t=m;
        while(t){
            if(t&1) b=b*a;
            t>>=1;
            a=a*a;
        }
        ull ans1=0;
        for(int i=sz;i<tot;i++)
            ans1=ans1+b.m[0][i];
        ans1--;

        Matrix c(2),d(2);
        d.setE();
        c.m[0][0]=26;c.m[0][1]=c.m[1][1]=1;
        while(m){
            if(m&1) d=d*c;
            m>>=1;
            c=c*c;
        }
        ull ans2=d.m[0][1]-1;
        return ans2-ans1;
    }
};
AC ac;
int main()
{
    ull n,m;   
    char s[10];
    while(scanf("%llu%llu",&n,&m)!=EOF){
        ac.init();
        for(int i=0;i<n;i++){
            scanf("%s",s);
            ac.inser(s);
        }
        ac.getfail();
        printf("%llu\n",ac.solve(m+1)); 
    }
}