CodeForces1337 E. Kaavi and Magic Spell (dp)

题意:

给两个串s和t,保证t串的长度不超过s串
初始你有一个空串a,
每次操作你可以取出s串的第一个字符,然后选择插入到a串的头部或者尾部

你可以不断进行操作直到s串被取完
问插入过程中有多少种可能的a串,满足t串是a串的前缀

答案对998244353取模

样例:
s串为:abab
t 串为:ba
在这里插入图片描述
红色表示的为满足条件的串

数据范围:
假设s串的长度为n,t串的长度为m
n<=3e3,m<=n

解法:

先用通配符补满t串直到长度为n(我直接用的0)

d[l][r]表示s串组合出t串的[l,r]的方案数
因为只关注s串能否匹配出t串的前m个,
所以通配符的位置可以用任意字符匹配

遍历s串依次插入,假设当前插入s的第i个字符
这时候结果串显然也是长度为i的串
枚举t串的左端点l,那么右端点r=l+i-1

如果s[i]==t[l],表明s[i]插入到[l+1,r]左边可以组成[l,r]
那么d[l][r]+=d[l+1][r]
如果s[i]==t[r],表明s[i]插入到[l,r-1]右边可以组成[l,r]
那么d[l][r]+=d[l][r-1]

最后的答案就是d[1][m]到d[1][n]的和

复杂度O(n^2)

code:

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=3e3+5;
const int mod=998244353;
char s[maxm],t[maxm];
int d[maxm][maxm];
int n,m;
signed main(){
    scanf("%s%s",s+1,t+1);
    n=strlen(s+1);
    m=strlen(t+1);
    for(int i=1;i<=n;i++){
        if(s[1]==t[i]||t[i]==0){
            d[i][i]=2;//因为第一个字符放左和放右算两种,因此是2而不是1
        }
    }
    for(int i=2;i<=n;i++){
        for(int l=1;l<=n;l++){
            int r=l+i-1;
            if(r>n)break;
            if(s[i]==t[l]||t[l]==0){//左边
                d[l][r]=(d[l][r]+d[l+1][r])%mod;
            }
            if(s[i]==t[r]||t[r]==0){//右边
                d[l][r]=(d[l][r]+d[l][r-1])%mod;
            }
        }
    }
    int ans=0;
    for(int i=m;i<=n;i++){
        ans=(ans+d[1][i])%mod;
    }
    cout<<ans<<endl;
    return 0;
}

发布了457 篇原创文章 · 获赞 37 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/weixin_44178736/article/details/105571730