题目
两种糖果分别有 9 个和 16 个,要全部分给 7 个小朋友,每个小朋友得到的糖果总数最少为 2 个最多为 5 个,问有多少种不同的分法。糖果必须全部分完
只要有其中一个小朋友在两种方案中分到的糖果不完全相同,这两种方案就算作不同的方案
分析
dfs
函数接受三个参数:depth
(表示当前的搜索深度)、n
(第一种糖果的剩余数量)、m
(第二种糖果的剩余数量)- 如果搜索深度达到 7,并且此时第一种糖果和第二种糖果的数量都为 0,就将结果
ans
加 1 并返回 - 然后通过两层循环遍历第一种糖果和第二种糖果可能选取的数量
i
和j
,条件是选取的总数i + j
在 2 到 5 之间。然后递归调用dfs
函数,更新深度和剩余糖果数量
代码
python
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)