CF908D New Year and Arbitrary Arrangement
一道期望dp,推方程不是很难,计算边界才是本题的难点。
我们A表示选a的概率,B表示选b的概率。
我们由于选了a的个数与后面有极大关系,同时正序状态无穷无尽,所以我们考虑倒推回来,因此我们设dp_{i,j}表示已经选了i个a,j个ab了后面选ab的个数的期望。
很显然有dp式:
dp_{i,j} = dp_{i+1,j} \times A + dp_{i,i+j} \times B
现在考虑一波边界,当j \leq k但k \leq i + j时,dp_{i,j}=i+j+ 0 \times B + 1 \times A \times B + 2 \times A^2 \times B + 3 \times A^3 \times B \dots
使用一波我不会的错位相减法可以求出dp_{i,j}=i+j+\frac{pa}{pb}
最后是答案,考虑到开头是b时不但可以自我无限转移,甚至还对答案无影响,所以我们考虑第一个字母直接为a,所以答案为dp_{1,0}
代码:
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
const LL mod = 1e9 + 7;
const int MAXN = 1005;
LL dp[MAXN << 2][MAXN << 2];
LL qpow(LL a , LL b) {
LL res = 1;
while(b) {
if(b & 1) res = res * a % mod;
a = a * a % mod;
b >>= 1;
}
return res;
}
int main() {
LL k , pa , pb;
scanf("%lld %lld %lld" , &k , &pa , &pb);
LL A = pa * qpow(pa + pb , mod - 2) % mod;
LL B = pb * qpow(pa + pb , mod - 2) % mod;
LL C = pa * qpow(pb , mod - 2) % mod;
for (int i = 1; i <= k * 2; ++i) {
for (int j = 0; j <= k * 2; ++j) {
if(i + j >= k) dp[i][j] = (i + j + C) % mod;
}
}
for (int i = k; i >= 1; --i) {
for (int j = k; j >= 0; --j) {
// if(i + j >= k) dp[i][j] = (i + j + C) % mod;
if(i + j >= k) continue;
dp[i][j] = (A * dp[i + 1][j] % mod + B * dp[i][i + j] % mod) % mod;
}
}
printf("%lld" , dp[1][0]);
return 0;
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· .NET Core内存结构体系(Windows环境)底层原理浅谈
· C# 深度学习:对抗生成网络(GAN)训练头像生成模型
· .NET 适配 HarmonyOS 进展
· .NET 进程 stackoverflow异常后,还可以接收 TCP 连接请求吗?
· SQL Server统计信息更新会被阻塞或引起会话阻塞吗?
· 传国玉玺易主,ai.com竟然跳转到国产AI
· 本地部署 DeepSeek:小白也能轻松搞定!
· 自己如何在本地电脑从零搭建DeepSeek!手把手教学,快来看看! (建议收藏)
· 我们是如何解决abp身上的几个痛点
· 普通人也能轻松掌握的20个DeepSeek高频提示词(2025版)