南京网预赛 11 BY bly

建个只有2个串的AC自动机,状态为dp[i][j][k][st]表示向右走了i步,向下走了j步,位于结点k,状态为st的总的方案数。

直接裸dp可过,一炮AC,比赛的时候这题看都没有看,比较可惜……主要是比赛前期出题要稳。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
typedef long long LL;
const LL mod=1000000007;
const int maxn=200;
const int sigma=2;
int ch[maxn][sigma],val[maxn],sz;
void init(){sz=1;memset(ch[0],0,sizeof(ch[0]));memset(val,0,sizeof(val));}
int idx(char c){return c=='R'?0:1;}
void insert(char *s,int v)
{
    int u=0,n=strlen(s);
    for(int i=0;i<n;i++)
    {
        int c=idx(s[i]);
        if(!ch[u][c])
        {
            memset(ch[sz],0,sizeof(ch[sz]));
            val[sz]=0;
            ch[u][c]=sz++;
        }
        u=ch[u][c];
    }
    val[u]|=(1<<v);
}
int f[maxn];
void getfail()
{
    queue<int> q;
    f[0]=0;
    for(int c=0;c<sigma;c++)
    {
        int u=ch[0][c];
        if(u) {f[u]=0;q.push(u);}
    }
    while(!q.empty())
    {
        int r=q.front();q.pop();
        for(int c=0;c<sigma;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];
            val[u]|=val[f[u]];
        }
    }
}
int m,n;
char s1[maxn],s2[maxn];
int dp[110][110][maxn][4];
int main()
{
    //freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d%d",&m,&n);
        scanf("%s%s",s1,s2);
        insert(s1,0);insert(s2,1);
        int l1=strlen(s1),l2=strlen(s2);
        getfail();
        for(int i=0;i<=m+1;i++)
            for(int j=0;j<=n+1;j++)
                for(int k=0;k<sz;k++)
                    for(int st=0;st<4;st++)
                        dp[i][j][k][st]=0;
        dp[0][0][0][0]=1;
        for(int i=0;i<=m;i++)
            for(int j=0;j<=n;j++)
                for(int k=0;k<sz;k++)
                    for(int st=0;st<4;st++) if(dp[i][j][k][st]>0)
                    {
                        dp[i+1][j][ch[k][0]][st|val[ch[k][0]]]=(dp[i+1][j][ch[k][0]][st|val[ch[k][0]]]+dp[i][j][k][st])%mod;
                        dp[i][j+1][ch[k][1]][st|val[ch[k][1]]]=(dp[i][j+1][ch[k][1]][st|val[ch[k][1]]]+dp[i][j][k][st])%mod;
                    }
        LL ans=0;
        for(int i=0;i<sz;i++) ans=(ans+dp[m][n][i][3])%mod;
        printf("%lld\n",ans);
        //printf("%I64d\n",ans);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值