京东笔试8-27号 最后一题(漂亮数)

输入一个n
输出长度为n的,只包括小写字母的,至少有两个red的字符串的数量。
解决方案:用总的字符串数量 减去 不包括一个red的,再减去包括一个red的字符串的数量
dp[i][j]表示前i个字符,其中第i个字符已经匹配到red的第j位了的字符串个数,例如dp[6][2]表示前6个字符,最后已经匹配到re了的字符串个数,即以re结尾的字符串个数。
代码如下:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
ll dp[N][4];
void init(int n){
	dp[0][0] = 1;
	for(int i = 1; i <= n; i ++){
		dp[i][0] = (dp[i - 1][0] * 25 % mod + dp[i - 1][1] * 24 % mod + dp[i - 1][2] * 24 % mod) % mod;
		dp[i][1] = (dp[i - 1][0] + dp[i - 1][1] + dp[i - 1][2]) % mod;
		dp[i][2] = dp[i - 1][1];
	}
}
int main()
{
	int n;
	cin >> n;
	ll ans = 1;
	for(int i = 1; i <= n; i ++){
		ans = ans * 26 % mod;
	}
	init(n);
	ans = (ans - (dp[n][0] + dp[n][1] + dp[n][2]) % mod + mod) % mod;
	for(int i = 1; i <= n - 2; i ++){
		ll left = (dp[i - 1][0] + dp[i - 1][1] + dp[i - 1][2]) % mod;
		ll right = (dp[n - i - 2][0] + dp[n - i - 2][1] + dp[n - i - 2][2]) % mod;
		ans = (ans - left * right % mod + mod) % mod;
	}
	cout << ans << endl;
	return 0;
}


#京东笔试##dp##笔试##京东#
全部评论
dp的ij没太看明白,可以举个例子吗大佬
点赞 回复 分享
发布于 2022-08-28 10:49 浙江

相关推荐

斑驳不同:还为啥暴躁 假的不骂你骂谁啊
点赞 评论 收藏
分享
11-09 01:22
已编辑
东南大学 Java
高级特工穿山甲:羡慕,我秋招有家企业在茶馆组织线下面试,约我过去“喝茶详谈”😢结果我去了发现原来是人家喝茶我看着
点赞 评论 收藏
分享
Pandaileee:校友加油我现在也只有一个保底太难了
点赞 评论 收藏
分享
6 13 评论
分享
牛客网
牛客企业服务