洛谷P1077 [NOIP2012 普及组] 摆花问题（详解+递归搜索树）

我们先来分析一下题目，这道题是典型的dp问题，也可以通过dfs来解决。

作为一个新手小白，我是先通过画出递归搜索树，用记忆化搜索（dfs）做的，然后根据dfs 得出dp的状态转移方程，改写为dp代码。

首先我们来带大家画一下递归搜索树（我觉得递归搜索树是一个很不错的解题思路）

方法一、记忆化搜索

#include <bits/stdc++.h>
using namespace std;
const int N= 105;
int n,m;
int arr[N];//数组记录每种类型花的数量。
int mem[N][N];//记忆化搜索用的数组。
int dfs (int x,int spm)//x代表花的类型，spm代表剩余的花需量。
{//dfs 搜索符合题目要求的方案数目。
    int ans=0;
    if (mem[x][spm])return mem[x][spm];//记忆化搜素
    if (spm==0)return 1;//根据递归搜索树，可以看出判断条件
    if (x>n||spm<0) return 0;//根据递归搜索树，可以看出判断条件
    for (int i=0;i<=arr[x];i++)//搜索树的分支（枚举每种类型的花选几盆）。
    {
        if (spm>=i)//一定要判断这个！！因为下面dfs中可能会出现形参为负。
      ans=(ans+dfs (x+1,spm-i))%1000007;//需要对每次结果进行取模操作
    }

    mem[x][spm]=ans;
    return ans;
}
int main ()
{
    cin>>n>>m;
    for (int i=1;i<=n;i++)
    {
        scanf ("%d",&arr[i]);
    }
    int ans=dfs(1,m);//递归搜索树顶端就是答案。
    cout << ans;
    return 0;
}

方法二、dp

#include <bits/stdc++.h>
using namespace std;
const int N=105;
int n,m;
int arr[N];
int f[N][N]={0};
int main ()
{
    cin>>n>>m;
    for (int i=1;i<=n;i++)
    {
        scanf ("%d",&arr[i]);
    }
    
    for (int i=1;i<=n;i++)
    {
        for (int j=0;j<=m;j++)
        {
           
            for (int z=0;z<=arr[i];z++)
            { if (j==0) f[0][0]=1;
            if (j>=z)
             f[i][j]=(f[i][j]+f[i-1][j-z])%1000007;
            }
        }
    }
    printf ("%d",f[n][m]);
    return 0;
}

方法三、优化dp（二维数组变成一维数组）（转化为01背包问题）

#include <bits/stdc++.h>
using namespace std;
const int N=105;
int n,m;
int arr[N];
int f[N]={0};
int main ()
{
    cin>>n>>m;
    for (int i=1;i<=n;i++)
    {
        scanf ("%d",&arr[i]);
    }
     f[0]=1;
    for (int i=1;i<=n;i++)
    {
        for (int j=m;j>=0;j--)
        {
           
            for (int z=1;z<=arr[i];z++)//注意z从1开始，因为一定会选择某种类型的花。
            {
            if (j>=z)
             f[j]=(f[j]+f[j-z])%1000007;
            }
        }
    }
    printf ("%d",f[m]);
    return 0;
}