各种球盒问题的python实现

最新推荐文章于 2023-08-13 16:14:14 发布

刘大鱼

最新推荐文章于 2023-08-13 16:14:14 发布

阅读量1.9k

点赞数 2

文章标签：球盒问题斯特林数 python

本文链接：https://blog.csdn.net/qq_40013071/article/details/89293652

版权

本文探讨了不同类型的球盒问题，包括球同盒同盒非空、盒可空等八种情况，并提供了斯特林数的理论背景和Python代码实现。通过递推公式和容斥原理详细解释了球不同盒同盒非空问题的解决方法，并展示了如何扩展到盒可空的情况。

摘要由CSDN通过智能技术生成

这几天终于有时间来记录一些有趣的问题了，是之前笔试的一道题的原型。没有系统的总结，对我来说还真的很难在笔试时候打出来（阅历不够啊）

本文基于：https://blog.csdn.net/zwz_511/article/details/46240927
里的理论进行一定补充及python实现

1. 球同，盒同，盒非空

P_m（N）的表示，m个同盒里放N个同球（非空）有多少种放法（因为盒同，所以[0,1,0],[0,0,1]…只算一种放法）。
运用递归。当盒数大于球数，是不可能成立的。

得到球与盒的差值（因为盒非空，所以每个盒先放进一个球）

如果目前盒的数量大于这个差值，那么就让盒数等于这个差值

如果盒等于1或0（这个0是为了防止原本和盒数和球数相同情况）返回1，代表这是1种。

盒为2，绝对不会超过球数的一半（5球2种，3球1种……）。

接着就是按公式的递归。
代码如下：

def ballandbox(box,ball):
    if box > ball:
        return -1
    ball2 = ball - box
    if box > ball2:
        box = ball2
    if box == 1 or box == 0:
        return 1
    if box == 2:
        return ball // 2
    result = 0
    for i in range(box):
        result += ballandbox(i+1,ball2)
    return result

模拟一遍。如4盒7球：
P₄（7） → box=4，ball =7

P₄（7）= P₁（3）+P₂（3）+P₃（3） → ball2 = 7-4=3，box = 3 进入迭代，result = 函数（1,3）+（2,3）+（3,3）

1 + 1 + 1 → box == 1，返回1，box == 2，返回3//2=1，box == 3，球数也为3，所以box == 0，返回1

2. 球同，盒同，盒可空

在1的基础上，该情况的公式是P_m（m+N）的表示，m个盒里放m+N个球（非空）有多少种放法。
意思是，N+m个球先把每个盒都放1个球，也就是放进m个球，剩下的N个球再放就相当于是盒可空的情况。
也就是在1的代码基础上，将输入改为m+N，并且当盒大于球数时，直接相当于盒等于球时的情况。

def ballandbox(box,ball):
    if box > ball:
        box = ball
    ball2 = ball - box
    if box > ball2:
        box = ball2
    if box == 1 or box == 0:
        return 1
    if box == 2:
        return ball // 2
    result = 0
    for i in range(box):
        result += ballandbox(i+1,ball2)
    return result

3.球同，盒不同，盒非空

高中题，插板法
box个盒子，ball个球。就相当于ball个球中间有ball-1个空，从这个些个空中挑出box-1个，剩下的类数为box个。对盒子的排序没有要求，所以公式也就是 $C_{box-1}^{ball-1}$

def main(box,ball):
    box = box - 1
    ball = ball -1
    return ballandbox3(box,ball)
    
def ballandbox3(box,ball):
    if ball == 1:
        return box
    elif box == 1 or ball == 0:
        return 1
    else:
        return ballandbox3(box-1