【HDU5201】一个有意思的容斥

1.题目链接。题目大意：n个桃子，m个猴子，猴子编号为1-m。其中猴王的序号是1，然后来分配这些桃子，要求1号猴王分配得到的桃子严格大于任意一只猴子，问有多少种分配方式。

2.一个比较经典的容斥问题。我们首先枚举1号得到桃子的个数1到n。然后考虑至少k个其他的猴子比这个猴王多，k从1到m-1.那么最终的结果就是至少0个-至少一个+至少两个.....。下面来看至少k个如何让计算：因为猴王此时为x个，至少k个猴子使得这个条件不成立的时候，我们从剩下的m-1个猴子拿出来k个，每个给他们x个，剩下的（n-(k+1)*x）个桃子，随意分配。在随意分配的过程中，我们知道就是这个模型：x1+x2...xn=m.其中n就是剩下的猴子，m就是剩下的桃子。上边那个方程对于x的非负数解的个数就是：C(m+n-1,m-1）。问题解决。

#include<bits/stdc++.h>
#pragma warning(disable:4996)
#define ll long long
using namespace std;
const ll N = 200005, mod = 1e9 + 7;
ll T, n, m, inv[N], fac[N], ans;
ll qpow(ll a, ll b)
{
	ll r = 1;
	while (b)
	{
		if (b & 1)
			r = r * a%mod;
		a = a * a%mod;
		b >>= 1;
	}
	return r;
}
ll C(ll n, ll m)
{
	return fac[n] * inv[n - m] % mod*inv[m] % mod;
}
int main()
{
	fac[0] = 1;
	for (int i = 1; i <= N - 5; i++)
		fac[i] = fac[i - 1] * i%mod;
	inv[N - 5] = qpow(fac[N - 5], mod - 2);
	for (int i = N - 6; i >= 0; i--)
		inv[i] = inv[i + 1] * (i + 1) % mod;
	scanf("%lld", &T);
	while (T--)
	{
		scanf("%lld%lld", &n, &m);
		if (m == 1 || n == 1)
		{
			puts("1");
			continue;
		}
		ans = 0;
		for (int x = 1; x <= n; x++)
			if (x > (n - x) / (m - 1))
				for (int k = 0; n - (k + 1)*x >= 0; k++)
					ans = (ans + ((k & 1) ? -1 : 1)*C(m - 1, k)*C(n - (k + 1)*x + m - 2, m - 2)) % mod;
		printf("%lld\n", (ans%mod + mod) % mod);
	}
	return 0;
}