HDU 2243 考研路茫茫——单词情结（AC自动机+矩阵快速幂）

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

---

WA：

1.单词的总数 = 26+26^2+26^3+......+26^L,而不是26^L

2.使用等比数列求和公式求单词总数：(26^(L+1)-26)/25,出现除法取模导致错误，这个25在模(2^64)下逆元我不会求.

---

做这题肯定做过POJ2778了吧

POJ 2778利用矩阵+AC自动机解决了走K步不出现一个给出单词的方法数。

而本题就需要用：单词总数  -  走1~L步不出现一个单词的总数

---

①先解决：求走1~L步不出现一个单词的总数

反正肯定不可能矩阵一遍遍乘然后每次求第0排的和

所以居然只要在原本矩阵右边添一列1就可以记录之前的了（不过要记得减一，因为初始化为1）

---

②单词总数：

f(i)表示26^1+....26^i

则有递推式:f(i) = f(i-1)*26+26

那么我们只需要做一个填字游戏：

 

从左往右从上到下依次填26,1,0,1

这个2X2矩阵进行矩阵快速幂L次然后用初始矩阵{0,26}乘这个矩阵即可算出总数

---

代码：

#include<bits/stdc++.h>

using namespace std;

#define ll long long
#define ull unsigned long long
#define for1(i,a,b) for (int i=a;i<=b;i++)
#define for0(i,a,b) for (int i=a;i<b;i++)
#define rof1(i,a,b) for (int i=a;i>=b;i--)
#define rof0(i,a,b) for (int i=a;i>b;i--)
#define pb push_back
#define fi first
#define se second
#define debug(x) printf("----Line %s----\n",#x)
#define pt(x,y) printf("%s = %d\n",#x,y)
#define INF 0x3f3f3f3f
#define dfl(x) ll x;scanf("%I64d",&x)
#define df2l(x,y) ll x,y;scanf("%I64d %I64d",&x,&y)
#define df(x) int x;scanf("%d",&x);
#define df2(x,y) int x,y;scanf("%d %d",&x,&y)
#define mod 1000000007
#define duozu(T) int T;scanf("%d",&T);while (T--)

const int N = 1e5+5;

const int maxnode = 30;//模式串数量*长度
const int ALP = 26;//字符种类数
char s[10];
ull mat[30][30];

struct AC_am
{
    queue<int>que;
    int sz;
    int trie[maxnode][ALP];
    int fail[maxnode];
    int last[maxnode];
    int val[maxnode];//储存当前节点信息，如是否为单词节点等等

    int newnode(int x){
        memset(trie[x],0,sizeof trie[x]);
        val[x] = 0;
        return sz++;
    }

    void init(){
        newnode(sz = 0);
    }

    int idx(char ch){//实际字符串转化为字典树对应节点，根据题目做出具体改变
        return ch-'a';
    }

    void insert(char *s){
        int u = 0;
        for (int i=0;s[i];i++){
            int c = idx(s[i]);
            if (!trie[u][c]){
                trie[u][c] = newnode(sz);
            }
            u = trie[u][c];
        }
        val[u]++;
    }

    void build(){
        fail[0] = 0;
        for (int c=0;c<ALP;c++){
            int v = trie[0][c];
            if (!val[v]) mat[0][v]++;
            if (v){
                que.push(v);
                fail[v] = 0;
                last[v] = 0;
            }
        }
        while (!que.empty()){
            int u = que.front();que.pop();
            for (int c=0;c<ALP;c++){
                int v = trie[u][c];
                if (!v){
                    trie[u][c] = trie[fail[u]][c];
                    v = trie[u][c];
                    if (!last[v] && !val[v]) mat[u][v]++;
                    continue;
                }
                int fa = fail[u];
                while (fa && !trie[fa][c]) fa = fail[fa];
                fail[v] = trie[fa][c];
                last[v] = val[fail[v]]? fail[v]:last[fail[v]];
                if (!last[v] && !val[v]) mat[u][v]++;
                que.push(v);
            }
        }
    }
}ac;

const int matsize = 30;
ull temp[matsize][matsize],ans[matsize][matsize];

int ms;
void multy(ull a[][matsize],ull b[][matsize])
{
    memset(temp,0,sizeof temp);
    for (int i=0;i<ms;i++)
        for (int j=0;j<ms;j++)
            for (int k=0;k<ms;k++)
                temp[i][j] += a[i][k]*b[k][j];
    for (int i=0;i<ms;i++)
        for (int j=0;j<ms;j++)
            a[i][j] = temp[i][j];
}

void matrix_fast_power(ull a[][matsize],ll b)
{
    memset(ans,0,sizeof ans);
    for (int i=0;i<ms;i++) ans[i][i] = 1;
    while (b){
        if (b&1) multy(ans,a);
        b>>=1;
        multy(a,a);
    }
    for (int i=0;i<ms;i++)
        for (int j=0;j<ms;j++)
            a[i][j] = ans[i][j];
}

ull fast_power(ull a,ll b)
{
    ull ans = 1;
    while (b){
        if (b&1) ans*=a;
        b>>=1;
        a*=a;
    }
    return ans;
}

int main()
{
    //freopen("C:/Users/DELL/Desktop/input.txt", "r", stdin);
    //freopen("C:/Users/DELL/Desktop/myoutput.txt", "w", stdout);
    int n;
    ll k;
    while (~scanf("%d %lld",&n,&k)){
        ac.init();
        memset(mat,0,sizeof mat);
        for0(i,0,n) scanf("%s",s),ac.insert(s);
        ac.build();
        ms = ac.sz;
        for (int i=0;i<=ms;i++) mat[i][ms] = 1;
        ms++;
        matrix_fast_power(mat,k);
        ull cnt = 0;
        for (int i=0;i<ms;i++) cnt += mat[0][i];
        ///求总数
        ull sum = 0;
        memset(mat,0,sizeof mat);
        mat[0][0] = 26,mat[0][1] = 1;
        mat[1][0] = 0,mat[1][1] = 1;
        ms = 2;
        matrix_fast_power(mat,k);

        sum = 0*mat[0][0] + 26*mat[0][1];

        ull ans = sum-(cnt-1);

        printf("%llu\n",ans);
    }
    return 0;
}