【洛谷 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$。

思路

搜索两次，一次输出方案总数，一次输出方案序列。

AC代码

#include <iostream>
#define AUTHOR "HEX9CF"
using namespace std;int a[15];
int n;
int cnt;
int flg;void f(int sum, int x, int y)
{a[y] = x;if (10 == y){if (sum + x == n){if (flg){for (int i = 1; i <= 10; i++){cout << a[i] << " ";}cout << endl;}else{cnt++;}}return;}for(int i = 1; i <= 3; i++) {f(sum + x, i, y + 1);}
}int main()
{cin >> n;f(0, 0, 0);flg = 1;cout << cnt << endl;f(0, 0, 0);return 0;
}