蓝桥杯-分糖果

"""
https://www.lanqiao.cn/problems/4124/learning/?page=1&first_category_id=1&problem_id=4124
"""
import os
import sys

# 记录方案数
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): # 当前小朋友的第二种糖果的数目
            # i + j表示当前第depth个小朋友分到的a, b糖果的个数
            if 2 <= i + j <= 5 and i <= n and j <= m:
                dfs(depth + 1, n - i, m - j)
dfs(0, 9, 16)
print(ans)