洛谷P1307 数字反转-蒲公英云

洛谷P1307 数字反转

题目背景

猪猪hanke得到了一只鸡

题目描述

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

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

输入输出格式

输入格式：

一行,n<=5000

输出格式：

第一行，方案总数

第二行至结束，10个数，表示每种配料所放的质量

按字典序排列。

如果没有符合要求的方法，就只要在第一行输出一个“0”

输入输出样例

输入样例#1：复制

输出样例#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

说明

枚举

直接贴代码，注释很详细

#include<stdio.h>
#include<iostream>
#include<cstring>
using namespace std;
int n;
int ans[10001][10];
int tmp[10];
int sum;
void dfs(int step, int all)                    //step表示当前是第几中调料，all表示当前的总调料量
{
    if (step == 10)                            //到了第10中就开始判断
    {
        if (all == n)                        //如果当前总调料量等于预定的n
        {
            int flag = 0;                    //flag=0表示10种调料里没有质量为0的，反之则有质量为0的
            for (int i = 0; i < 10; i++)
            {
                if (tmp[i] != 0)
                {
                    ans[sum][i] = tmp[i];
                }
                else
                {
                    flag = 1;
                    break;
                }
            }
            if (!flag)
                sum++;                        //没有质量为0的调料就方案数加1
        }
        return;
    }
    else
    {
        for (int j = 1; j <= 3; j++)//每种调料的三种重量
        {
            if (all + j > n) break; //如果此时大于预定的调料，就退出不勇计算后面的调料
            all += j;                //总调料量加上当前这种调料的质量
            tmp[step] = j;            //记录
            dfs(step + 1, all);        //接着下一种调料计算
            tmp[step] = 0;            //置为零，表示这种调料尝试过了这种情况然后去掉
            all -= j;                //相应的总调料量减去当前这种的调料的量
        }
    }
}
int main()
{
    cin >> n;
    dfs(0, 0);
    if (!sum)
        cout << 0;
    else
    {
        cout << sum;
        cout << '\n';
        for (int i = 0; i < sum; i++)
        {
            for (int j = 0; j < 10; j++)
            {
                cout << ans[i][j] << ' ';
            }
            cout << '\n';
        }
    }
    return 0;
}