微信公众号怎么创建步骤网站关键词优化排名公司
题目传送门
前言
每次在做数学的时候(尤其是组合排列),都深深地感受到一种无力感 . . . . . . ...... ......一种看完题解感觉自己跟傻*一样地无力感 . . . . . . ...... ......
思路
我们先考虑最多会剩下多少个空房间:
k k k 个人各向左走一步就会剩下 k k k 个空房间,但是由于 k k k 可能大于等于 n n n,所以 k k k 要与 n − 1 n - 1 n−1 取 m i n min min。
然后我们枚举剩下 i i i 个空房间, i ∈ [ 0 , m i n ( k , n − 1 ) ] i \in [0, min(k, n - 1)] i∈[0,min(k,n−1)],每个有 C n i C_n^i Cni 种选择。
然后就是把 n n n 个人划分到 n − i n - i n−i 个房间里的可能数:
这就相当于在 n n n 个无差别的人之间插入隔板,将其分成 n − i n - i n−i 个非空集合,就有 C n − 1 n − i − 1 C_{n - 1}^{n - i - 1} Cn−1n−i−1 种可能。
所以一共就有: ∑ i = 0 m i n ( k , n − 1 ) C n i × C n − 1 n − i − 1 \sum_{i = 0} ^ {min(k, n - 1)} C_n^i \times C_{n - 1} ^ {n - i - 1} ∑i=0min(k,n−1)Cni×Cn−1n−i−1 种可能。
代码
#include <bits/stdc++.h>using namespace std;typedef long long ll;const int maxn = 2e5 + 7;
const int mod = 1e9 + 7;ll n, k, ans;
ll fac[maxn], inv[maxn];
ll qpow(ll x, ll y) {ll res = 1;for (; y; y >>= 1, x = x * x % mod)if (y & 1) res = res * x % mod;return res;
}
ll C(ll x, ll y) {return fac[x] * inv[x - y] % mod * inv[y] % mod;}
int main() {scanf("%lld%lld", &n, &k);fac[0] = inv[0] = 1;for (ll i = 1; i <= n; ++i) {fac[i] = fac[i - 1] * i % mod;inv[i] = qpow(fac[i], mod - 2);}for (int i = 0; i <= min(k, n - 1); ++i)ans = (ans + C(n, i) * C(n - 1, i) % mod) % mod; // C(n - 1, n - i - 1) = C(n - 1, i)printf("%lld\n", ans);return 0;
}