蓝桥杯 Python 研究生组-2023-省赛-分糖果-深度算法

题目

两种糖果分别有 9 个和 16 个，要全部分给 7 个小朋友，每个小朋友得到的糖果总数最少为 2 个最多为 5 个，问有多少种不同的分法。糖果必须全部分完 

只要有其中一个小朋友在两种方案中分到的糖果不完全相同，这两种方案就算作不同的方案

分析

• dfs 函数接受三个参数：depth（表示当前的搜索深度）、n（第一种糖果的剩余数量）、m（第二种糖果的剩余数量）
• 如果搜索深度达到 7，并且此时第一种糖果和第二种糖果的数量都为 0，就将结果 ans 加 1 并返回
• 然后通过两层循环遍历第一种糖果和第二种糖果可能选取的数量 i 和 j，条件是选取的总数 i + j 在 2 到 5 之间。然后递归调用 dfs 函数，更新深度和剩余糖果数量

代码

ans=0
def dfs(depth,n,m):
    global ans
    if depth==7:
        if n==0 and m==0:
            ans+=1
            return
    for i in range(0,n+1):#第一种糖果
        for j in range(0,m+1):#第二种糖果
            if i+j<=5 and i+j>=2:
                dfs(depth+1,n-i,m-j)
dfs(0,9,16)
print(ans)