列表

详情


NC25538. 华华跟奕奕玩游戏

描述

有一个箱子,开始时有n个黑球,m个蓝球。每一轮游戏规则如下:
第一步:奕奕有p的概率往箱子里添加一个黑球,有(1-p)的概率往箱子里添加一个蓝球。
第二步:华华随机从箱子里取出一个球。
华华喜欢黑球,他想知道k轮游戏之后箱子里黑球个数的期望。

输入描述

输入五个整数n,m,k,a,b。
1<=n,m<=1e6,1<=k<=1e9
其中p=,且a<=b,0<=a<1e9+7,0<b<1e9+7

输出描述

输出一个数表示k轮游戏后箱子里黑球个数的期望。
输出一个整数,为答案对1e9+7取模的结果。即设答案化为最简分式后的形式为,其中a和b互质。输出整数 x 使得bx≡a(mod 1e9+7)且0≤x<1e9+7。可以证明这样的整数x是唯一的。

示例1

输入:

2 2 1 1 2

输出:

2

示例2

输入:

2 2 2 3 10

输出:

184000003

原站题解

上次编辑到这里,代码来自缓存 点击恢复默认模板

C++11(clang++ 3.9) 解法, 执行用时: 3ms, 内存消耗: 504K, 提交时间: 2020-08-04 16:33:15

#include <cstdio>
#define mod 1000000007

int n,m,k,a,b;
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;x=1ll*x*x%mod,y>>=1);return re;}
#define inv(x) ksm(x,mod-2)

int main()
{
	scanf("%d %d %d %d %d",&n,&m,&k,&a,&b);
	int p=1ll*a*inv(b)%mod,T=1ll*(n+m)*inv(n+m+1)%mod;
	printf("%d",(1ll*ksm(T,k)*n%mod+1ll*p*T%mod*(ksm(T,k)-1+mod)%mod*inv(T-1)%mod)%mod);
}

pypy3(pypy3.6.1) 解法, 执行用时: 46ms, 内存消耗: 31084K, 提交时间: 2020-08-03 20:31:25

MOD = 1000000007
def fpow(a, b, p=1000000007):
  return pow(a, b, p)

n, m, k, a, b = map(int, input().split())
p = a * fpow(b, MOD - 2) % MOD
s = (n + m) * fpow(n + m + 1, MOD - 2) % MOD
ans = n * fpow(s, k) + p * (s - fpow(s, k + 1) + MOD * 100) * fpow(1 - s, MOD - 2)
ans %= MOD
print(ans)

Python3(3.5.2) 解法, 执行用时: 26ms, 内存消耗: 3552K, 提交时间: 2020-08-03 20:32:50

n, m, k, a, b = map(int, input().split())
mod = 10**9+7
m0 = pow(n+m, k, mod)
m1 = pow(n+m+1, k, mod)
left = b*m1
right = b*n*m0+a*(n+m)*(m1-m0)
ans = pow(left, mod-2, mod)*right
print(ans % mod)

上一题