0818 T1 flower

原题链接

题目分析

一道简单的组合数问题，学过数学选修2-3的同学应该都知道怎么计算组合数。
c(m, n) = m! / ((m-n)!*n!)
以上就是公式。

优化

但是我们注意到n, m最大能到5,0000， 所以直接套公式是不可行的。
我们可以在每一步阶乘的时候进行因数分解。
将每个数因数分解后加到总答案里，这样就会大大节省空间。
具体实现参考代码。

代码

 #include<cstdio>
 #include<cmath>
using namespace std;
const int maxn = 50000 + 55; 
int a[maxn], n, m, r;
long long p, ans;

void work(int n, int d) { //对n^d进行因数分解 
    r = (int)(sqrt(n) + 1e-5);//对n开方，减少误差 
    for(int i = 2; i <= r; ++i) {
        while(n%i == 0) {
            n /= i;
            a[i] += d;
        }
    }//对n质因数分解 
    if(n > 1) a[n] += d;
}

int main() {
    scanf("%d%d%lld", &n, &m, &p);
    for(int i = 2; i <= n; ++i) work(i, 1);
    for(int i = 2; i <= m; ++i) work(i, -1);
    for(int i = 2; i <= n-m; ++i) work(i, -1);
    ans = 1;
    for(int i = 2; i <= n; ++i)
        for(int j = 1; j <= a[i]; ++j)
            ans = ans * i % p;
    printf("%lld\n", ans);
    return 0;
}