CF86C Genetic engineering

最新推荐文章于 2020-10-24 16:19:29 发布

C202044zxy

最新推荐文章于 2020-10-24 16:19:29 发布

阅读量404

点赞数

分类专栏： AC自动机 dp

本文链接：https://blog.csdn.net/C202044zxy/article/details/104455799

版权

dp 同时被 2 个专栏收录

94 篇文章 0 订阅

订阅专栏

AC自动机

17 篇文章 0 订阅

订阅专栏

一、题目

点此看题

二、解法

可以把题目给的限制当成模式串覆盖文本串，最后要求文本串的每个位置都被覆盖，结合套路，我们定义 $f [i] [j] [k]$ 为还剩长度 $i$ ，当前匹配到的点是 $j$ ，还未覆盖的一段后缀长度为 $k$ 的方案数。

转移枚举下一位字符，我们对于 $\text{AC}$ 自动机上的每个点维护一个最大匹配长度（ $f a i l$ 链上的），如果跳到新点的长度大于还未覆盖的后缀长度，那么清零还未覆盖的后缀长度，否则用后缀长度 $+ 1$ 继续跑。

边界条件是后缀长度 $\geq10$ （此时已无力回天）返回 $0$ ，文本串跑完了看还未匹配的后缀长度，是 $0$ 返回 $1$ ，否则返回 $0$ ，贴个代码。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
const int M = 105;
const int MOD = 1e9+9;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,m;
char s[M];
struct Automaton
{
    int c[M][4],val[M],fail[M],cnt;
    int dp[1005][M][11];
    void ins(char *s)
    {
        int len=strlen(s),now=0;
        for(int i=0; i<len; i++)
        {
            int v=0;
            if(s[i]=='A') v=0;
            if(s[i]=='C') v=1;
            if(s[i]=='T') v=2;
            if(s[i]=='G') v=3;
            if(!c[now][v]) c[now][v]=++cnt;
            now=c[now][v];
        }
        val[now]=len;
    }
    void build()
    {
        queue<int> q;
        for(int i=0; i<4; i++) if(c[0][i]) q.push(c[0][i]);
        while(!q.empty())
        {
            int t=q.front();
            q.pop();
            val[t]=max(val[t],val[fail[t]]);
            for(int i=0; i<4; i++)
                if(c[t][i]) fail[c[t][i]]=c[fail[t]][i],q.push(c[t][i]);
                else c[t][i]=c[fail[t]][i];
        }
        memset(dp,-1,sizeof dp);
    }
    int dfs(int x,int p,int w)
    {
        if(w>=10) return 0;
        if(x==0) return w==0;
        if(dp[x][p][w]!=-1) return dp[x][p][w];
        dp[x][p][w]=0;
        for(int i=0; i<4; i++)
        {
            int to=c[p][i];
            dp[x][p][w]=(dp[x][p][w]+dfs(x-1,to,(val[to]>w)?0:w+1))%MOD;
        }
        return dp[x][p][w];
    }
} AC;
signed main()
{
    n=read();
    m=read();
    for(int i=1; i<=m; i++)
    {
        scanf("%s",s);
        AC.ins(s);
    }
    AC.build();
    printf("%d\n",AC.dfs(n,0,0));
}