G、subsequence 1

牛客多校第五场

题目链接

大概意思就是给两个串,一个长串一个短串,问长串中有多少个子序列形成的数字大于短串

我们首先可以确定长串中位数大于短串的子序列的个数(不以0开头),长度为N的串中找到长度为M的子序列的个数为：

 即从N个字符中选出M个字符

而长度相等的子序列我们该怎找呢。

定义dp[i][j]代表长串中1~i中长度为j的子序列与短串中1~j 相等的个数

如果当前位s[i] == s1[j] 那么dp[i][j] = dp[i - 1][j] + dp[i - 1][j - 1];

如果当前位s[i] > s1[j]那么说明在1~i - 1中选出长度为 j - 1并于s1中1~j- 1相等的子序列加上这个字符,再从剩下的 n - i 个字符中选出

m - j 个字符构成的子序列一定大于s1串,所以答案直接加上就可以。

显然,dp[0][0] = dp[i][0] = 1;

代码：

#include <bits/stdc++.h>
using namespace std;
const int mod = 998244353;
int c[3005][3005];
void init(){
    c[0][0] = 1;
    for(int i = 0; i <= 3000; i ++){
        c[i][0] = 1;
        for(int j = 1; j <= i; j++){
            c[i][j] =(c[i - 1][j - 1] + c[i - 1][j])%mod;
        }
    }
}
int dp[3005][3005];
int main()
{
    init();

    int t;
    cin >> t;
    while(t --){

        int ans = 0;
        int n,m;
        cin >> n>>m;
        string s,s1;
        cin >> s >> s1;

        for(int i = 0; i < s.size() - s1.size(); i ++){
            if(s[i] != '0'){
                for(int j = s1.size(); j <= s.size() - i - 1; j ++){
                    if(i + j + 1> s.size()) break;
                    ans = (ans + c[s.size() - i - 1][j])%mod;
                }
            }
        }
        s = " " + s;
        s1 = " " + s1;
        dp[0][0] = 1;//dp[i][j]表示s串1~i为中的子序列长度为j与s1串中1-j相等的串的个数
        for(int i = 1; i <= n; i ++){
            dp[i][0] = 1;
            for(int j = 1; j <= m; j ++){
                dp[i][j] = dp[i - 1][j];
                if(s[i] == s1[j]) dp[i][j] = (dp[i][j] + dp[i - 1][j - 1])%mod;
                else if(s[i] > s1[j]) ans = (ans + 1LL*dp[i - 1][j - 1]*c[n - i][m - j])%mod;// 前面 1 ~ i- 1 的长度为j - 1子序列的个数*在后面n - i个数中选 m - j 个字符构成字序列都大于s1串
            }

        }

        cout << ans << endl;
    }
}