Codeforces 696D Legen...(AC自动机+矩阵快速幂)

http://codeforces.com/problemset/problem/696/D

题意:给你n个串,每个串有个开心值,然后让你构造一个长为l的串,使得开心值之和最大。
题解:AC自动机处理这些串,然后开心值加在终止节点上,可能同一个节点有多个串,开心值就要加起来,然后bfs建立trie图的时候,要把失配之后的节点的开心值也要加到这个节点上,这样就不会漏掉了。
然后对于矩阵的每个节点i,j,表示走一步的时候,从i状态走到j状态,获得的开心值,直接就是AC自动机里的开心值就行了。矩阵初始化要为-INF,表示无法走到,0表示可以走到但是没有获得开心值,正数就是可以走到并且这里有终止节点。
然后就是跑矩阵快速幂,这个矩阵快速幂和以前的不同,每次跑得时候,是求 c[i][j]=max(c[i][j],a[i][k]+b[k][j]) 这样,表示这步可以取到的最大开心值。所以矩阵快速幂的ans矩阵也和以前不同


代码

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           10005
#define   MAXN          1000005
#define   maxnode       205
#define   sigma_size    26
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-4;
const LL    mod    = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

struct AC{  
    int ch[maxnode][sigma_size];  
    LL val[maxnode];
    int last[maxnode], f[maxnode];  
    int sz;  
    void init(){  
        mem(ch[0],0);  
        last[0]=val[0]=0;  
        sz = 1;  
    }  
    int idx(char c){return c-'a';}  
    void insert(const char *s,int v){  
        int n = strlen(s), u=0;  
        for(int i=0;i<n;i++){  
            int c = idx(s[i]);  
            if(!ch[u][c]){  
                mem(ch[sz],0);  
                val[sz]=0;  
                ch[u][c]=sz++;  
            }  
            u = ch[u][c];  
        }  
        val[u]+=v;  
    }  
    void bfs(){  
        queue<int>q;  
        f[0]=0;  
        for(int c = 0; c<sigma_size;c++){  
            int u = ch[0][c];  
            if(u){q.push(u); f[u]=last[u]=0;}  
        }  
        while(!q.empty()){  
            int r =q.front(); q.pop();  
            val[r]+=val[f[r]];
            for(int c = 0;c<sigma_size;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];  
                last[u] = val[f[u]] ? f[u] : last[f[u]];  
            }  
        }  
    }  
}ac;  

struct Matrix{  
    int n;  
    LL maze[maxnode][maxnode];  

    void init(int n){  
        this->n = n;  
        mem(maze,0);  
    } 
    Matrix operator + (Matrix &rhs){  
        Matrix m;  
        m.init(n);  
        for(int i=0;i<n;i++){  
            for(int j=0;j<n;j++){
                m.maze[i][j]=-1e18;
                for(int k=0;k<n;k++){
                    m.maze[i][j]=max(m.maze[i][j],maze[i][k]+rhs.maze[k][j]);
                }
            }  
        }  
        return m;  
    }  
};  

Matrix qpow(Matrix A,LL n){
    if(n==1) return A;
    Matrix ans=A;
    n--;
    while(n){
        if(n&1) ans=ans+A;
        A=A+A;
        n>>=1;
    }
    return ans;
}
int a[205];
char s[205];
int main(){
    //freopen("in.txt","r",stdin);
    int n;
    LL l;
    while(cin>>n>>l){
        ac.init();
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        for(int i=0;i<n;i++){
            scanf("%s",s);
            ac.insert(s,a[i]);
        }
        ac.bfs();
        Matrix A;
        A.init(ac.sz);
        for(int i=0;i<ac.sz;i++){
            for(int j=0;j<ac.sz;j++) A.maze[i][j]=-1e18;
        }
        for(int i=0;i<ac.sz;i++){
            for(int j=0;j<26;j++){
                int p=ac.ch[i][j];
                A.maze[i][p]=ac.val[p];
            }
        }
        A=qpow(A,l);
        LL ans=0;
        for(int i=0;i<A.n;i++){
            ans=max(ans,A.maze[0][i]);
        }
        cout<<ans<<endl;
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值