递归/DFS学习4(洛谷：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\le 5000$。

思路：

这题是要在固定的10个位置填入相应的数，所以可以使用dfs来做，是指数型枚举（每个位置的选数与其他位置没有任何影响，不要求按照什么顺序，什么升序要求之类的），很明显美味程度要在10~~30之间，不然没有方案的。我们对每个位置的美味指数从1到3开始枚举，因为要先输出方案，所以可以用二维数组把符合要求的方案存储，先输出方案数，再输出各个方案。注意剪枝（如果加起来大于要求，直接return），不然会超时。

正确代码：

#include <bits/stdc++.h>
using namespace std;
int n, res = 0;
vector<int> ans(11);
vector<vector<int>> v(1e5,vector<int>(11));//二维数组的定义方式
void dfs(int x, int sum) {
	if(sum>n){//剪枝
		return;
	}
	if (x > 10) {
		if (sum == n) {
			res++;
			for (int i = 1; i <= 10; i++) {
				v[res][i]=ans[i];
			}
		}
		return;
	}
	for (int j = 1; j <= 3; j++) {
		ans[x] = j;
		dfs(x + 1, sum + j);
	}
}

int main() {
	cin >> n;
	if(n<10||n>30){
		cout << 0;
		return 0;
	}
	dfs(1, 0);
	cout << res << '\n';
	for(int i=1;i<=res;i++){
		for(int j=1;j<=10;j++){
			cout << v[i][j] << ' ';
		}
		cout << '\n';
	}
	return 0;
}

c++ stl代码（学习c++容器后）

#include <bits/stdc++.h>
using namespace std;
int n;
vector<vector<int>> ans;
vector<int> re;
void dfs(int cnt,int val){
	if(cnt==10){
		if(val==n){
			ans.push_back(re);
		}
		return;
	}
	for(int i=1;i<=3;i++){
		re.push_back(i);
		dfs(cnt+1,val+i);
		re.pop_back();
	}
}
int main()
{
	cin >> n;
	dfs(0,0);
	cout << ans.size() << '\n';
	for(int i=0;i<ans.size();i++){
		for(int j=0;j<10;j++){
			cout << ans[i][j] << " \n"[j==9];
		}
	}
}

学到的东西：

vector二维数组的创建：vector<vector> v(1e5,vector(11));