计算多项式系数

题目描述
给定一个多项式$(ax+by{)}^k$，求$a^n{b}^m$系数

输入格式
共一行,包含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;
}