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

avatar
作者
猴君
阅读量:0

题目

两种糖果分别有 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)

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!