hdu 5894 hannnnah_j’s Biological Test

题目链接

题意：

$n$个座位要安排$m$个人，每两个人之间至少隔$k$个位置，求方案数。

题解：

因为$m$个人之间有$m$个空隙，我们设第$i$空隙为$a_i$，因此有$a_1+a_2+\cdots +a_m=n-m$，这里$a_i\ge k$，我们可以改写成$b_1+b_2+\cdots +b_m=n-m-m\ast k$，这里$b_i\ge 0$，我们将等式右边的$m$可以通过给等式左边的每个数$+1$消去，得$c_1+c_2+\cdots +c_m=n-m\ast k$，其中$c_i\ge 1$，这样就可以使用插板法求出总的方案数，是$C_{n-m\ast k-1}^{m-1}$。

由于总共有$n$个位置，因此方案数要乘上$n$，不过由于这里每个人没有区别，且我们之前求出得正整数解通过移动每组解总共有$m$组解相同，比如$x_1=1,x_2=2,x_3=3$其实和$x_1=3,x_2=1,x_3=2$相同。因此我们得答案还要再除以$m$。即$\frac{C_{n-m\ast k-1}^{m-1}\ast n}{m}$。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e6 + 10;
const int mod = 1e9 + 7;
ll f[MAXN], inv[MAXN];
ll qpow(ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1) {
			ans = ans * a % mod;
		}
		a = a * a % mod;
		b >>= 1;
	}
	return ans;
}
void init() {
	f[0] = 1;
	inv[0] = 1;
	for (ll i = 1; i <= 1000000; i++) {
		f[i] = f[i - 1] * i % mod;
		inv[i] = qpow(f[i], mod - 2) % mod;
	}
}
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int t;
	init();
	cin >> t;
	while (t--) {
		ll n, m, k;
		cin >> n >> m >> k;
		if (m == 1) { // 只安排一个学生
			cout << n << endl;
			continue;
		}
		if (n < m * k + m) {
			cout << 0 << endl;
			continue;
		}
		ll ans = f[n - m * k - 1] * inv[m - 1] % mod * inv[n - m * k - m] % mod * n % mod * qpow(m, mod - 2) % mod;
		cout << ans << endl;
	}
	return 0;
}