洛谷p2089烤鸡-DFS深度优先搜索

思路打开

于 2024-04-10 23:19:50 发布

阅读量545

点赞数 9

文章标签：深度优先算法

本文链接：https://blog.csdn.net/m0_74276681/article/details/137615527

版权

文章描述了一个编程问题，要求根据给定的美味程度n，找出10种配料（每种可选1-3克）的所有可能组合，使得总和等于n。使用DFS（深度优先搜索）算法进行求解，并在满足条件时输出方案总数及配料质量的排列。

摘要由CSDN通过智能技术生成

题目描述

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

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

一个正整数 n，表示美味程度。
输出格式

第一行，方案总数。

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

如果没有符合要求的方法，就只要在第一行输出一个 0。
输入
11
输出
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

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 20;
int n;
int a[N];   //存方案
int ret = 0; //存方案数 
int arr[59055][N];  // 总排列数 = 10的3次方 = 59050
void dfs(int x, int sum) {
	if(sum > n) return;  //剪枝 
	
	if(x > 10) {
		if(sum == n) {
			ret ++;
			for(int i = 1; i <= 10; i ++) {
				arr[ret][i] = a[i];
				}
			}
			return;
		}
	
	for(int i = 1; i <= 3; i ++) {
		a[x] = i;
		dfs(x + 1,sum + i);
		a[x] = 0;
	}
}

int main() {
	 cin >> n;
	 dfs(1,0);
	 cout << ret;
	 cout << endl;
	 for(int i = 1; i <= ret; i ++) {
	 	for(int j = 1; j <= 10; j ++) {
	 		cout << arr[i][j];
		 }
		 cout << endl;
	 }
  return 0;
}