【洛谷 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"
#define f(i, x) for (int i = 1; i <= x; i++)
using namespace std;

int em(int n, int flg){
    int cnt;
    int q, w, e, r, t, y, u, i, o, p;
    f(q, 3)
        f(w, 3)
            f(e, 3)
                f(r, 3)
                    f(t, 3)
                        f(y, 3)
                            f(u, 3)
                                f(i, 3)
                                    f(o, 3)
                                        f(p, 3)
    {
        if (n == q + w + e + r + t + y + u + i + o + p)
        {
            if(flg){
                cout << q << " " << w << " " << e << " " << r << " " << t << " " << y << " " << u << " " << i << " " << o << " " << p << endl;
            }
            cnt++;
        }
    }
    return cnt;
}

int main()
{
    int n;
    cin >> n;
    cout << em(n, 0) << endl;
    em(n, 1);
    return 0;
}