一、题目
题目描述:
中秋节公司分月饼,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)) # 输出分月饼方法数