多重集下的组合数

多重集定义

多重集是指包含重复元素的广义集合
例如 S = { n 1 ⋅ a 1 , n 2 ⋅ a 2 , . . . , n k ⋅ a k } S=\{n_1\cdot a_1,n_2\cdot a_2,...,n_k\cdot a_k\} S={n1​⋅a1​,n2​⋅a2​,...,nk​⋅ak​}就是由$n_1$个$a_1$，$n_2$个$a_2$，…，$n_k$个$a_k$组成的多重集

---

多重集的排列数

给定一个多重集 S = { n 1 ⋅ a 1 , n 2 ⋅ a 2 , . . . , n k ⋅ a k } S=\{n_1\cdot a_1,n_2\cdot a_2,...,n_k\cdot a_k\} S={n1​⋅a1​,n2​⋅a2​,...,nk​⋅ak​}
其排列数为
$$A=\frac{N!}{n_1!n_2!...n_k!},(N=\sum _{i=1}^k{n}_i)$$

---

多重集的组合数

（Q1）
给定一个多重集 S = { n 1 ⋅ a 1 , n 2 ⋅ a 2 , . . . , n k ⋅ a k } S=\{n_1\cdot a_1,n_2\cdot a_2,...,n_k\cdot a_k\} S={n1​⋅a1​,n2​⋅a2​,...,nk​⋅ak​}
以及一个整数$m$，且有$m<=n_i(\forall i\in [1,k])$
求从S中取出任意m个元素可以产生的不同多重集数量
（A1）
$ans=C_{k+m-1}^{k-1}$

证明
题目所求就是满足${\sum }_{i=1}^k{x}_i=m$的多重集 { x 1 ⋅ a 1 , x 2 ⋅ a 2 , . . . , x k ⋅ a k } \{x_1\cdot a_1,x_2\cdot a_2,...,x_k\cdot a_k\} {x1​⋅a1​,x2​⋅a2​,...,xk​⋅ak​}的数量
换句话说
就是就是要给每个$x_i$赋值$[0,m]$中的任意整数，且所有$x_i$的和为m
(这里因为有$m<=n_i(\forall i\in [1,k])$，所以不用考虑$x_i>n_i$的情况)

若我们用0的个数表示每个$x_i$的值
那么我们可以转化原问题为
用$k-1$个$1$将$m$个$0$分为$k$组的方案数
也就是多重集 { m ⋅ 0 , ( k − 1 ) ⋅ 1 } \{m\cdot 0,(k-1)\cdot 1\} {m⋅0,(k−1)⋅1}的排列数
故得
$$ans=\frac{(m+k-1)!}{m!(k-1)!}=C_{k+m-1}^m=C_{k+m-1}^{k-1}$$

---

（Q2）
给定一个多重集 S = { n 1 ⋅ a 1 , n 2 ⋅ a 2 , . . . , n k ⋅ a k } S=\{n_1\cdot a_1,n_2\cdot a_2,...,n_k\cdot a_k\} S={n1​⋅a1​,n2​⋅a2​,...,nk​⋅ak​}
以及一个整数$m(m<={\sum }_{i=1}^k{n}_i$)
求从S中取出任意m个元素可以产生的不同多重集数量
（A2）
$$ans=C_{k+m-1}^{k-1}-\sum _{i=1}^k{C}_{k+m-n_i-2}^{k-1}+\sum _{1<=i,j<=k,i!=j}{C}_{k+m-n_i-n_j-3}^{k-1}-...+(-1{)}^k{C}_{k+m-{\sum }_{i=1}^k{n}_i-(k+1)}^{k-1}$$

证明
上面的式子显然用到了容斥定理
怎么理解呢

先不考虑$n_i$得限制
那么从其中取m个元素的方案数为$C_{k+m-1}^{k-1}$

这里的方案数显然包含了$x_i>n_i$的不合法方案
设$S_i$表示至少包含$n_i+1$个$a_i$的多重集(也就是$a_i$个数不合法的多重集)
我们需要从上述总方案数中减去$S_i$的数量

怎么求$S_i$的数量呢
我们先从$S$中取出$n_i+1$个$a_i$
再在$S$任选$m-n_i-1$个元素
即构成了$S_i$
根据Q1证明可得其数量为$C_{k+m-n_i-2}^{k-1}$

减去了所有只考虑$x_i>n_i$的不合法方案后
会发现我们还多减去了$S_i\bigcap S_j$的数量
(也就是同时包含了至少$n_i+1$个$a_i$和至少$n_j+1$个$a_j$的多重集)
我们再把这部分数量加回来${\sum }_{1<=i,j<=k,i!=j}{C}_{k+m-n_i-n_j-3}^{k-1}$

反复容斥下去即可得到上述式子
$$ans=C_{k+m-1}^{k-1}-\sum _{i=1}^k{C}_{k+m-n_i-2}^{k-1}+\sum _{1<=i,j<=k,i!=j}{C}_{k+m-n_i-n_j-3}^{k-1}-...+(-1{)}^k{C}_{k+m-{\sum }_{i=1}^k{n}_i-(k+1)}^{k-1}$$

---

将上述推导实际应用
CodeForces - 451E Devu and Flowers【多重集下的组合数+容斥】题解