2019牛客国庆集训派对day1 第B题 组合数 解题报告

试题

试题链接

题目描述

给出 n 和 k，求 $\min \left\{\frac{n!}{k!(n-k)!},10^{18}\right\}$ 的值。
其中 $n!=1\times 2\times \cdots \times n$， 表示 n 的阶乘。

输入描述

输入文件包含多组数据，请处理到文件结束。
每组数据包含两个整数 n 和 k。

• $0⩽k⩽n⩽10^9$
• 至多 $10^5$ 组数据。

输出描述

对于每组数据，输出一个整数，表示所求的值。

样例输入

1000000000 0
1000000000 2
1000000000 500000000

样例输出

1
499999999500000000
1000000000000000000

解题思路

首先，根据杨辉三角形，可知
$$C_m^n=C_m^{m-n}$$

所以，$C_{10000}^{9001}$ 可以弄成 $C_{10000}^{999}$ 来算，这样子就快多了。

然后，我们注意到组合数公式：
$$C_m^n=\frac{A_m^n}{n!}=\frac{m!}{(m-n)!n!}$$

再看排列数公式：
$$A_n^m=\frac{m!}{(m-n)!}=m\bullet (m-1)\bullet (m-2)\bullet \cdots \bullet (m-n+1)$$

例如，
$$A_9^4=9\times 8\times 7\times 6$$

对比以上两个式子，发现了什么问题呢？

对，组合数公式实际上是可以化简分步进行的。

$$\begin{array}{rcl}{C}_m^n& =& \frac{m\bullet (m-1)\bullet (m-2)\bullet \cdots \bullet (m-n+1)}{n\bullet (n-1)\bullet (n-2)\bullet \cdots \bullet 1}\\ & & \\ & =& \frac{m}{1}\bullet \frac{m-2}{2}\bullet \frac{m-3}{3}\bullet \cdots \bullet \frac{m-n+1}{n}\end{array}$$

例如，

$$C_9^4=\frac{9}{1}\times \frac{8}{2}\times \frac{7}{3}\times \frac{6}{4}$$

这个计算可以循环，也就是

1. result 赋值为1
2. 第一轮循环，result 乘上 $\frac{9}{1}$
3. 第二轮循环，result 乘上 $\frac{8}{2}$
4. 第三轮循环，result 乘上 $\frac{7}{3}$
5. 第四轮循环，result 乘上 $\frac{6}{4}$

那么现在，应该知道怎么编写 $C_m^n$ 的算法了。

源代码

#include <cstdio>

const long long upper_limit = 1000000000000000000LL;

long long C(long long m, long long n)
{
    if (n == 0 || n == m)
        return 1;
    if (n > m / 2)
        n = m - n;
    __int128 res = (__int128)1;
    for (long long i = 1; i <= n; ++i)
    {
        res = res * (m-i+1) / i;
        if (res > upper_limit)
            return upper_limit;
    }
    return (long long)res;
}

int main()
{
    long long n, k;
    while (scanf("%lld%lld", &n, &k) != EOF)
    {
        printf("%lld\n", C(n, k));
    }
    return 0;
}