Codeforces Round #362 (Div. 2) F. Legen...(AC自动机+矩阵快速幂)

题意:
我们有n(200)个特殊单词,总长度不超过200。
我们要构造一个长度为l(1e14)的字符串。
如果字符串每包含一个给定单词i(可重叠匹配),则我们的权值计数+a[i]
问你最大可以得到的权值


思路:
对自动机上的状态,我们可以建立状态之间的转移可以获得的权值
->对建立好之后的矩阵跑快速幂


#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int MOD=1e9+7;
const int N=210;
const int G=202;

struct Matrix{
    long long m[N][N];

    Matrix operator *(const Matrix &b)const{
        Matrix ans;
        for(int i=0;i<G;i++)
            for(int j=0;j<G;j++){
                ans.m[i][j]=-1e18;
                for(int k=0;k<G;k++)
                    ans.m[i][j]=max(ans.m[i][j],m[i][k]+b.m[k][j]);
            }
        return ans;
    }
};

Matrix base;

Matrix Pow(long long n){
    Matrix ans;
    for(int i=0;i<G;i++)
        for(int j=0;j<G;j++)
            ans.m[i][j]=-1e18;
    ans.m[0][0]=0;
    while(n){
        if(n&1) ans=ans*base;
        n>>=1;
        base=base*base;
    }
    return ans;
}

struct Trie{
    int next[N][26],fail[N],sz,root;
    long long val[N];

    int newnode(){
        memset(next[sz],-1,sizeof(next[sz]));
        sz++;
        return sz-1;
    }

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

    void insert(char s[],int cnt){
        int u=root;
        for(int i=0;s[i];i++){
            int c=s[i]-'a';
            if(next[u][c]==-1)
                next[u][c]=newnode();
            u=next[u][c];
        }
        val[u]+=cnt;
    }

    void build(){
        queue<int>Q;
        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 u=Q.front();
            Q.pop();
            for(int i=0;i<26;i++){
                if(next[u][i]==-1)
                    next[u][i]=next[fail[u]][i];
                else{
                    fail[next[u][i]]=next[fail[u]][i];
                    val[next[u][i]]+=val[ fail[next[u][i]] ];
                    Q.push(next[u][i]);
                }
            }
        }
        for(int i=0;i<G;i++)
            for(int j=0;j<G;j++)
                base.m[i][j]=-1e18;
        for(int i=0;i<sz;i++)
            for(int j=0;j<26;j++)
                base.m[i][next[i][j]]=val[next[i][j]];
    }

};
Trie ac;
int a[N];
char s[N];

int main(){
    int n;
    long long m;
    ac.init();
    scanf("%d%lld",&n,&m);
    for(int i=1;i<=n;i++)   scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        ac.insert(s,a[i]);
    }
    ac.build();
    Matrix tmp=Pow(m);
    long long ans=0;
    for(int i=0;i<ac.sz;i++)
        ans=max(ans,tmp.m[0][i]);
    printf("%lld\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值