题目描述
给定一个多项式
(
a
x
+
b
y
)
k
(ax+by)^k
(ax+by)k,求
a
n
b
m
a^nb^m
anbm系数
输入格式
共一行,包含5个整数,分别为a,b,k,n,m每两个整数之间用一个空格隔开.
输出格式
出共1行,包含一个整数,表示所求的系数,这个系数可能很大,输出对1000000007取模后的结果.
注意:k最大取到1000000
思路:如果k规模较小,可以利用杨辉三角打表。
k规模较大时只能利用公式。
对幂的求解采用快速幂算法。
#include <iostream>
using namespace std;
const int mod = 1000000007;
long long f[1000001];
//阶乘取模
void init() {
for (long long i = 0; i < 1000001; i++) {
if (!i || i == 1)
f[i] = 1;
else
f[i] = f[i - 1] * i % mod;
}
}
//快速幂取模
long long qpow(long long a, long long n) {
long long ans = 1;
while (n) {
if (n & 1)
ans = ans * a % mod;
a = a * a % mod;
n >>= 1;
}
return ans;
}
int main() {
init();
long long a, b, k, n, m, t;
cin >> t;
for (int i = 0; i < t; i++) {
cin >> a >> b >> k >> n >> m;
cout << "case #" << i << ":\n";
long long res = qpow(a, n) * qpow(b, m) % mod;
long long res2 = f[k] * qpow(f[m] * f[k - m] % mod, mod - 2) % mod;
res = res * res2 % mod;
cout << res << endl;
}
return 0;
}