CodeForces - 697F Legen... AC自动机+快速幂

题目链接点这里

这题很好，很强大，做了一晚上。。。。

如果对这题毫无思路的，建议先坐下这道题。做完这道题，结合10^14的数据，应该可以想到本题也应该利用AC自动机构造状态转移矩阵，然后利用快速幂加速

不过这里的矩阵乘法定义为  M[i][j]=max(a[i][k]+b[k][i],M[i][j])；

#include<iostream>
#include<cstdio>
#include<math.h>
#include<algorithm>
#include<map>
#include<set>
#include<bitset>
#include<stack>
#include<queue>
#include<string.h>
#include<cstring>
#include<vector>
#include<time.h>
#include<stdlib.h>
using namespace std;
#define INF 0x3f3f3f3f
#define INFLL 0x3f3f3f3f3f3f3f3f
#define FIN freopen("input.txt","r",stdin)
#define mem(x,y) memset(x,y,sizeof(x))
typedef unsigned long long ULL;
typedef long long LL;
#define MX 1111111
const LL p=10000000000;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<pair<int,int>,int> PIII;
typedef pair<int,int> PII;
int n;
LL l;
struct Matrix
{
    LL M[222][222],n;//改
    Matrix () {}
    Matrix(int _n,LL x)  //初始化
    {
        n=_n;
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++) M[i][j]=x;
    }
    Matrix operator *(const Matrix &a) const
    {
        Matrix ret=Matrix(n,-1);
        for(int i=0; i<n; i++)
            for(int k=0; k<n; k++)
                for(int j=0; j<n; j++)
                    ret.M[i][j]=max(ret.M[i][j],((a.M[i][k]!=-1&&M[k][j]!=-1)?a.M[i][k]+M[k][j]:-INF));
        return ret;
    }
};
Matrix pow_Matrix(Matrix a,LL x)
{
    Matrix ret=Matrix(a.n,0);
    // for(int i=0; i<a.n; i++) ret.M[i][i]=0;
    while(x)
    {
        if(x&1) ret=ret*a;
        a=a*a;
        x>>=1;
    }
    return ret;
}
char s[1000+10];
struct AC_trie
{
    int root,cnt;
    int End[5000+10],to[5000+10][27],fail[5000+10];//改
    int newnode()
    {
        for(int i=0; i<26; i++) to[cnt][i]=-1;//改
        End[cnt]=0;
        return cnt++;
    }
    void init()
    {
        cnt=0;
        root=newnode();
    }
    void insert(char *s,int w)
    {
        int now=root;
        for(char *ss=s; *ss; ss++)
        {
            int v=*ss-'a';//改
            if(to[now][v]==-1)to[now][v]=newnode();
            // cout<<*ss<<" "<<to[now][v]<<endl;
            now=to[now][v];

        }
        End[now]+=w;//改

    }
    void build()
    {
        queue<int> my;
        fail[root]=root;
        for(int i = 0; i < 26; i++)//改
            if(to[root][i] == -1)
                to[root][i]= root;
            else
            {
                fail[to[root][i]]=root;
                my.push(to[root][i]);
            }
        //
        while(!my.empty())
        {
            int u=my.front();
            End[u]+=End[fail[u]];
            //cout<<u<<" "<<fail[u]<<" "<<End[u]<<" "<<End[fail[u]]<<endl;
            my.pop();
            for(int i=0; i<26; i++)   //改
            {
                int v=to[u][i];
                if(v!=-1)
                {
                    fail[v]=to[fail[u]][i];
                    my.push(v);
                }
                else to[u][i]=to[fail[u]][i];
            }
        }
    }
    Matrix Matrix_build()
    {
        Matrix a=Matrix(cnt,-1);
        for(int i=0; i<cnt; i++)
            for(int j=0; j<26; j++) a.M[i][to[i][j]]=End[to[i][j]];
        return a;
    }
} AC;
int val[222];
int main()
{
    FIN;
    while(~scanf("%d%I64d",&n,&l))
    {
        AC.init();
        for(int i=1; i<=n; i++) scanf("%d",&val[i]);
        for(int i=1; i<=n; i++)
        {
            scanf("%s",s);
            AC.insert(s,val[i]);
        }
        AC.build();
        Matrix a=AC.Matrix_build();
        // for(int i=0; i<AC.cnt; i++)
        //   for(int j=0; j<AC.cnt; j++) printf("%I64d%c",a.M[i][j],j==AC.cnt-1?'\n':' ');
        Matrix b=pow_Matrix(a,l);
        // for(int i=0; i<AC.cnt; i++)
        //  for(int j=0; j<AC.cnt; j++) printf("%I64d%c",b.M[i][j],j==AC.cnt-1?'\n':' ');
        LL ans=0;
        for(int i=0; i<AC.cnt; i++)ans=max(ans,b.M[0][i]);
        printf("%I64d\n",ans);
    }
    return 0;
}