【洛谷 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;
}