ZOJ 3557 隔板法 + Lucas 定理

题意

传送门 ZOJ 3557 How Many Sets II

题解

求从编号为 $1\cdots N$ 的 $N$ 个元素中选取 $M$ 个元素，且元素编号不相邻的方案数。

元素不相邻，那么选取的元素间至少存在一个未选取的元素。将未选取元素看作隔板，则答案为 $C_{N-M+1}^M$。由于 $C_n^m=A_n^m/m!$，则可以 $O(M)$ 暴力求解组合数。为了防止出现无法求解逆元的情况，即 $P|M!$，应用 $Lucas$ 定理求解组合数。

#include <bits/stdc++.h>
using namespace std;
#define rep(i, l, r) for (int i = l, _ = r; i < _; ++i)
typedef long long ll;
int N, M, P;

int pow_mod(int x, int n)
{
    int res = 1 % P;
    x %= P;
    while (n)
    {
        if (n & 1)
            res = (ll)res * x % P;
        x = (ll)x * x % P, n >>= 1;
    }
    return res;
}

int C(int n, int m)
{
    if (m == 0)
        return 1;
    if (n < m)
        return 0;
    int a = 1, b = 1;
    rep(i, n - m + 1, n + 1) a = (ll)a * i % P;
    rep(i, 1, m + 1) b = (ll)b * i % P;
    return (ll)a * pow_mod(b, P - 2) % P;
}

int Lucas(int n, int m)
{
    if (m == 0)
        return 1;
    if (n < m)
        return 0;
    return (ll)C(n % P, m % P) * Lucas(n / P, m / P) % P;
}

int main()
{
    while (cin >> N >> M >> P)
    {
        cout << Lucas(N - M + 1, M) << '\n';
    }
    return 0;
}