2021杭电多校第七场 Link with Balls

2021杭电多校第七场

Link with Balls

题目链接

题意

有2n个桶(每个桶里面都有无数个球)，要从这些桶里面取出m个球，问有多少种取球方案。

取球的限制为：

• 在$2x-1$号桶里面取出$kx$个球($k$可以为任意非负整数)。
• 在$2x$号桶中取出at most x个球。

思路

首先考虑每个桶能取的个数：

桶的$id$	1($2\times 1-1$)	2($2\times 1$)	3($2\times 2-1$​)	4($2\times 2$​)	5($2\times 3-1$​)	6($2\times 3$​)
可以取的球数	$k\times 1$	$1$	$k\times 2$	$1,2$	$k\times 3$	$1,2,3$

可以发现：

$id=1$​​​的桶中可以取出任意个球。

如果将$id=2$的桶和$id=3$的桶看成一个桶，也可以取出任意个球$(2\times k+1)or(2\times k)$)。

如果将$id=4$的桶和$id=5$的桶看成一个桶，也可以取出任意个球$(3\times k+2)or(3\times k+1)or(3\times k)$。

$id=n=2\times 3$的桶比较特殊，只能取出$x\in [1,n]$个球。

通过上面的转换，原本的$2n$​个桶转变成n​个可以取出任意数球的桶和1个可以取$x\in [1,n]$​个球的桶P。

如果$n\le m$：

从$0\Rightarrow n$枚举P桶取球的个数t，然后$m-t$的球由其他$n$个桶取，用隔板法可以求出值为$C_{(m-t-1)+n}^{n-1}$。（如果不会隔板法可以看我另一个题解，里面有说。链接）

但是这道题目的数据很多($T=10^5$​​)，所以不能直接遍历，考虑将式子化简。

由公式$C_a^b=C_{a-1}^b+C_{a-1}^{b-1}\Rightarrow C_{a-1}^{b-1}=C_a^b-C_{a-1}^b$：
$$\begin{array}{rl}\sum _{t=0}^n{C}_{m-t-1+n}^{n-1}& =C_{m+n-1}^{n-1}+C_{m+n-2}^{n-1}+...+C_{m-1}^{n-1}\\ & =(C_{m+n}^n-C_{m+n-1}^n)+(C_{m+n-1}^n-C_{m+n-2}^n)+...+(C_m^n-C_{m-1}^n)\\ & =C_{m+n}^n-C_{m-1}^n\end{array}$$

如果$n>m$​​​：

$$\begin{array}{rl}\sum _{t=0}^n{C}_{m-t-1+n}^{n-1}& =C_{m+n-1}^{n-1}+C_{m+n-2}^{n-1}+...+C_{n-1}^{n-1}\\ & =(C_{m+n}^n-C_{m+n-1}^n)+(C_{m+n-1}^n-C_{m+n-2}^n)+...+(C_{n+1}^n-C_n^n)+C_{n-1}^{n-1}\\ & =C_{m+n}^n-C_n^n+C_{n-1}^{n-1}\\ & =C_{m+n}^n\end{array}$$

AC的代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
// =============================================组合数板子
const int N = 2e6 + 10, mod = 1e9 + 7;
ll fac[N + 10];
ll mul(ll a, ll b) {
	return ((a * b - (long long)((long long)((long double)a / mod * b + 1e-3) * mod)) % mod + mod) % mod;
}
ll power(ll a, ll b) {
    ll res = 1;
    while (b) {
        if (b & 1) res = mul(res, a);
        b >>= 1;
        a = mul(a, a);
    }
    return res;
}
ll inv(ll x) { return power(x, mod - 2); }
// 预处理
ll prepare() {
    fac[0] = 1;
    for (int i = 1; i <= N; i++) {
        fac[i] = mul(fac[i - 1], i);
    }
}
// 求组合数
ll C(ll n, ll m) {
    if (m > n) return 0;
    return mul(mul(fac[n], inv(fac[m])), inv(fac[n - m]));
}
// ===================================================

int main() {
    prepare();
    ll t, n, m;
    scanf("%lld", &t);
    while (t--) {
        scanf("%lld%lld", &n, &m);
        if (n <= m)
            printf("%lld\n", (C(m + n, n) - C(m - 1, n) + mod) % mod);
        else
            printf("%lld\n", C(m + n, n));
    }

    return 0;
}