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
代码如下:
/*
整理题目条件关键词如下:
10种配料
要求所有配料质量之和等于美味程度n
输出所有搭配方案
按照字典序
*/
#include <iostream>
#include <vector>
//使用stl库中向量(动态)数组前提
#include <algorithm>
//包含stl中常用算法查找算法(std::find)排序算法(std::sort)数值算法(std::accumulate)
using namespace std;
//引入std的命名空间,由于c++标准库中(入函数,类,对象等)都定义在这个名称空间中
//引入后,后就不用再std::前缀,比如没引入前std::cout<<引入后 只需写cout<<
int n;//美味程度
vector<int> res;//存储某一次搜索方案
vector<vector<int>> memo;//存储整个搜索过程中的所有方案
int cnt = 0;//存储最终方案数量
void print(vector<int> res) {
for (int i = 0; i < res.size(); i++)
cout << res[i] << " ";
cout << endl;
}
//打印方案
void dfs(int index, int quantity) {
//index当前已经选到哪种配料
//quantity当前美味程度
if (index == 11) {
//此时判断所有配料是否选择
if (quantity == n)
//还要符合美味程度
{
vector<int> temp = res;
cnt++;
memo.push_back(temp);
}
return;
}
if (quantity > n)return;//剪枝
//没有判断到最后一种配料,美味程度就已经达到了,后面就不需要继续搜索了
for (int i = 1; i <= 3; i++) {
//本质是组合问题,每一次选择可以是重复的元素,如都选择1作为质量
res.push_back(i);
//选择质量为i的配料
dfs(index + 1, quantity + i);
//搜索下一种配料(也就是处理下一种配料)
//此时递归深度逐渐向下
//不选他,则选择下一种配料,回溯,回到上一层递归,将上一层递归中选择的结果删除,重新选择
res.pop_back();
}
}
int main() {
cin >> n;
dfs(1, 0);
//dfs标识符号常用表示深度优先搜索(或者是深度优先搜索的模板)
//从第1中配料开始选择,此时美味程度是0,由于还没确定选不选第一种配料
cout << cnt << endl;
for (const auto& row : memo) {
//二维数组的结构化绑定
for (int value : row) {
cout << value << " ";
}
cout << endl;
}
return 0;
}
//深度优先搜索使用递归算法实现,与数据结构“栈”类似
//递归深度不断增加,就是搜索
//递归深度减小,就叫回溯
//回溯过程中,会产生结果
//使用递归的可能因为数据巨大(递归执行次数过多)导致程序超时,可以使用
//剪枝操作:提前结束递归,记忆化搜索:保存之前递归的结果,当某两次递归发生条件相同时,返回记忆化数组中的结果,减少继续搜索的时间
//使用递归搜索树进行梳理递归程序运行逻辑