洛谷[普及]:P2089 烤鸡

烤鸡

感谢题目提供者叫我※森哥

题目背景

猪猪 Hanke 得到了一只鸡。

题目描述

猪猪 Hanke 特别喜欢吃烤鸡（本是同畜牲，相煎何太急！）Hanke 吃鸡很特别，为什么特别呢？因为他有 10 种配料（芥末、孜然等），每种配料可以放 1 到 3 克，任意烤鸡的美味程度为所有配料质量之和。

现在， Hanke 想要知道，如果给你一个美味程度 n ，请输出这 10 种配料的所有搭配方案。

输入格式

一个正整数 n，表示美味程度。

输出格式

第一行，方案总数。

第二行至结束，10 个数，表示每种配料所放的质量，按字典序排列。

如果没有符合要求的方法，就只要在第一行输出一个 0。

样例 #1

样例输入 #1

11

样例输出 #1

10
1 1 1 1 1 1 1 1 1 2 
1 1 1 1 1 1 1 1 2 1 
1 1 1 1 1 1 1 2 1 1 
1 1 1 1 1 1 2 1 1 1 
1 1 1 1 1 2 1 1 1 1 
1 1 1 1 2 1 1 1 1 1 
1 1 1 2 1 1 1 1 1 1 
1 1 2 1 1 1 1 1 1 1 
1 2 1 1 1 1 1 1 1 1 
2 1 1 1 1 1 1 1 1 1

提示/提示

对于 100% 的数据，n <= 5000。

题解

# total        当前总重量
# a            当前处理的配料索引，从0开始
# m            当前的配料组合列表
# results      存储所有有效配料组合的列表

def ingredients(total,a,m,results):

    global kind                    # 在Python中，如果想要递归函数修改一个全局变量，你需要在函数内部使用 global 关键字声明
    if a == 10:                    # 已经为所有10种配料分配了重量
        if total == n:             # total 是否等于用户输入的美味程度n
            results.append(m[:])   # 将m中从头到尾(0到9)的数据放到results
            kind += 1
        return

# 应该检查是否大于 n 而不是大于等于，因为继续添加配料会导致总重量超过要求

    elif total >= n:       
        return

# 如果以上条件都不满足，说明还有配料需要分配重量
# 对于当前的配料（索引为 a），从1到3克进行循环，为每种可能的重量分配，并递归调用 ingredients 函数，同时更新 total、a 和 m

    else:
        for i in range(1,4):
            m[a] = i
            ingredients(total + i,a + 1,m,results)

n = int(input())
kind = 0
results = []                
m = [0] * 10                # 初始化配料数组
ingredients(0,0,m,results)
print(kind)                 # 打印总的配料组合数量

# 使用 sorted 函数对 results 列表进行排序，确保输出的配料组合按字典序排列
# 使用 map(str, combination) 将每个整数配料组合转换为字符串，并用空格连接起来，以便打印

for _ in sorted(results):   
    print(' '.join(map(str,_))) # 打印每个配料组合

---

Python 中的 sorted() 函数是一个内置函数，用于对可迭代对象的元素进行排序，并返回一个新的排好序的列表。这个函数不会修改原始的可迭代对象。

以下是 sorted() 函数的一些关键点：

1. 使用方式：sorted(iterable, key=None, reverse=False)。

• iterable：要排序的可迭代对象。
• key：一个函数，它会被用来在进行比较之前从每个列表元素中提取一个比较键（比如通过 key=str.lower 来实现不区分大小写的排序）。
• reverse：布尔值。如果设置为 True，则列表元素将被逆序排列，默认为 False。

2. 返回值：sorted() 函数返回一个列表，该列表是输入的可迭代对象的元素按照升序排列的新列表。

3. 稳定性：sorted() 函数是稳定的排序算法，这意味着相等的元素的顺序在排序后会保持不变。

4. 类型：可以对任何可迭代对象进行排序，包括列表、元组、字典、集合等。

5. 复杂度：sorted() 函数通常使用 TimSort 算法，该算法在最坏情况下的时间复杂度为 O(n log n)。

6. 示例：

# 对列表进行排序
numbers = [3, 1, 4, 1, 5, 9, 2, 6]
sorted_numbers = sorted(numbers)
print(sorted_numbers)                        # 输出: [1, 1, 2, 3, 4, 5, 6, 9]

# 对字典进行排序（默认按字典的键排序）
my_dict = {'banana': 3, 'apple': 4, 'pear': 1}
sorted_items = sorted(my_dict.items())       # 返回一个按字典键排序的列表
print(sorted_items)                          # 输出: [('apple', 4), ('banana', 3), ('pear', 1)]

# 使用 key 参数进行排序
words = ['banana', 'Apple', 'Cherry']
sorted_words = sorted(words, key=str.lower)  # 忽略大小写的排序
print(sorted_words)                          # 输出: ['Apple', 'banana', 'Cherry']

在烤鸡配料问题中，sorted(results) 用于对所有找到的配料组合进行排序，确保输出是按字典序排列的。这样，输出的配料组合列表会按照第一个元素从小到大排序，如果第一个元素相同，则按照第二个元素排序，依此类推。

---

print(' '.join(map(str,_)))

• map(str,_)：map 函数在这里用于将 _ (列表中的每个整数)转换成字符串
  • str 是Python的内置函数，用于将对象转换成字符串
  • map 函数接收两个参数
    • 第一个是函数（在这里是 str）
    • 第二个是可迭代对象（在这里是 _）
  • map 函数会对 _ (每个元素)执行 str 函数，返回一个迭代器，其中的元素是转换后的字符串。
• ' '.join(...)：join 是字符串的内置方法，用于将序列中的元素以指定的分隔符连接生成一个新的字符串。在这里，分隔符是空格 ' '。join 方法被调用来连接 map 函数返回的迭代器中的所有字符串元素。

举个例子，假设列表是 [1, 2, 3]，那么这行代码的执行流程如下：

1. map(str, [1, 2, 3]) 将返回迭代器 ['1', '2', '3']。
2. ' '.join(['1', '2', '3']) 将迭代器中的字符串元素用空格连接起来，生成字符串 '1 2 3'。
3. print('1 2 3') 将字符串 '1 2 3' 打印到控制台。