【华为机试】2023年真题B卷（python）-分月饼

一、题目

题目描述：

中秋节公司分月饼，m个员工，买了n个月饼，m<=n，每个员工至少分1个月饼，但可以分多个，单人份到最多月饼的个数为Max1，单人分到第二多月饼的个数是Max2，Max1-Max2<=3。
同理，单人分到第n-1多月饼的个数是Max(n-1)，单人分到第n多月饼的个数是Max(n)，Max(n-1)-Max(n)<=3。
请问有多少种分月饼的方法？

二、输入输出

输入描述：
第一行输入m n，表示m个员工，n个月饼，m<=n
输出描述：
输出有多少种月饼分法

三、示例

示例1：
输入：
2 4
输出：
2
说明：
4 = 1+3  和3+1算一种分法
4 = 2+2  共两种方案

四、解题思路

可以通过递归不断缩小问题规模，将大问题拆分为更小的子问题，并通过递归调用来解决子问题。同时，为了避免重复计算相同的参数组合，使用了装饰器来添加了缓存功能，提高了代码的执行效率。

五、参考代码 

# -*- coding: utf-8 -*-
'''
@File    :   2023-B-分月饼.py
@Time    :   2023/12/19 14:02:56
@Author  :   mgc 
@Version :   1.0
@Desc    :   None
'''# import os
# import re
# import sys
# import copy
# import math
# import queue
# import functools
# from queue import Queue
# from collections import Counter, defaultdictimport functools@functools.lru_cache
def calc_count(n, m, last_max):"""计算分月饼的方法数Args:n (int): 剩余月饼数量m (int): 剩余员工数量last_max (int): 上一个员工分到的月饼的最大数量Returns:int: 分月饼的方法数"""if m == 0 and n == 0:return 1if n < 0 or m < 0:return 0comb_count = 0moon_cakes = [last_max + i for i in range(4)]for mc in moon_cakes:if mc * m > n:breakcomb_count += calc_count(n - mc, m - 1, mc)return comb_countdef cal(m, n):"""计算分月饼的方法数Args:m (int): 员工数量n (int): 月饼数量Returns:int: 分月饼的方法数"""if m == n:return 1else:comb_count = 0for i in range(1, n - m + 2):if m * i > n:breakcomb_count += calc_count(n - i, m - 1, i)return comb_countm, n = map(int, input().split())  # 输入m个员工和n个月饼print(cal(m, n))  # 输出分月饼方法数