牛客多校第五场
大概意思就是给两个串,一个长串一个短串,问长串中有多少个子序列形成的数字大于短串
我们首先可以确定长串中位数大于短串的子序列的个数(不以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;
}
}