牛客——等级之题 N1（8.3）

链接：https://ac.nowcoder.com/acm/contest/6766/A
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld

题目描述

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

输入描述:

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

输出描述:

输出一个数表示k轮游戏后箱子里黑球个数的期望。
输出一个整数，为答案对1e9+7取模的结果。即设答案化为最简分式后的形式为\frac{a}{b}
b
a
​
,其中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

粘一下牛客大佬写的题解，很易理解

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
const int N = 2e5 + 500;
ll quick(ll a,ll b) {
	ll ans = 1;
	while(b) {
		if(b & 1)ans = ans * a % mod;
		b >>= 1;
		a = a * a % mod;
	}
	return ans;
}
int main() {
	ll n,m,k,a,b;
	scanf("%lld %lld %lld %lld %lld",&n,&m,&k,&a,&b);
	ll A = 1ll * (n + m) * quick(n + m + 1,mod - 2) % mod;
	ll B = 1ll * (n + m) * a % mod * quick(b,mod - 2) % mod * quick(n + m +1,mod - 2) % mod;
	B = B % mod; 
	ll ans = quick(A,k) * n % mod + B * (1 - quick(A,k) + mod) % mod * quick(1 - A,mod - 2) % mod;
	ans = (ans % mod + mod) % mod;
	printf("%lld\n",ans);
	return 0;
}