NC25538. 华华跟奕奕玩游戏
描述
输入描述
输入五个整数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)