【NC16596】计算系数

最新推荐文章于 2024-10-01 23:05:47 发布

木又可可

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量465

点赞数 6

分类专栏： # 数学文章标签：算法

本文链接：https://blog.csdn.net/m0_52319522/article/details/137642272

版权

数学专栏收录该内容

18 篇文章 0 订阅

订阅专栏

本文介绍了如何通过快速幂算法和卢卡斯定理计算给定条件下的组合数，并给出了C(k,m)与两个幂的乘积模10007的结果。提供了C++代码示例来解决数学题目的计算问题。

摘要由CSDN通过智能技术生成

题目

计算系数

组合数，快速幂

思路

这是一道数学题，由之前的数学知识可以知道，题目要我们算一个数： $C_k^ma^nb^m)\mod\ 10007$

题意很明显，没有弯弯绕，就是要求这个数！可以知道，这个数由一个组合数和两个幂相乘，幂运算很好求，使用快速幂即可，组合数的求法有很多，这里记录卢卡斯定理（见字符串）

代码

#include <stdio.h>

#define MOD 10007

typedef long long LL;

// 老朋友：快速幂
int quick_pow(int a, int b, int p) {
    int res = 1;
    while (b) {
        if (b & 1) {
            res = (LL)res * a % p;
        }
        a = (LL)a * a % p;
        b >>= 1;
    }
    return res;
}

// 求 a 在模 p 条件下的逆元，要求 a 与 p 互质
int inv(int a, int p) { return quick_pow(a, p - 2, p); }

// 卢卡斯定理求组合数 C(n,m) % p 的值
int lucas(int n, int m, int p) {
    if (!m) return 1;
    int ni = n % p, mi = m % p;
    if (ni < mi) return 0;
    int res = 1;
    for (int i = 0; i < mi; i++) {
        res = (LL)res * (ni - i) % p * inv(i + 1, p) % p;
    }
    return res * lucas(n / p, m / p, p) % p;
}

int main(void) {
    int a = 0, b = 0, k = 0, n = 0, m = 0;
    scanf("%d%d%d%d%d", &a, &b, &k, &n, &m);
    printf("%d\n", (lucas(k, m, MOD) * quick_pow(a, n, MOD)) % MOD *
                       quick_pow(b, m, MOD) % MOD);
    return 0;
}