题目描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第ii种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。试编程计算,一共有多少种不同的摆花方案。
输入格式
第一行包含两个正整数n和m,中间用一个空格隔开。第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1,a2,…,an
输出格式
一个整数,表示有多少种方案。
注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
输入输出样例:
输入
2 4
3 2
输出
2
(这道题完全可以用动态规划,所以我为什么要用搜索???)
这道题用搜索很好写,不过一定要去做标记,也就是记忆化搜索,何为记忆化搜索?就是把搜索过的记下来,省得再费力气再搜一次
这道题呢,我觉得值得注意的是 dfs 里面的 if 判定语句顺序一定不能乱,当 b (代表的是已经用到的盆数)等于目标是直接加一,不往下继续执行了,判断是否标记过也要放在最后,防止重复,因为可能到最后一位时只需要加一就可以退出执行了,并不需要重复的再加一遍
#include <bits/stdc++.h>
using namespace std;
int n,m;
int A[109];
int mod=1000007;
int B[109][109];
int dfs (int a,int b)
{
if(b>m) return 0;
if(b==m) return 1;
if(a==n+1) return 0;
if(B[a][b]) return B[a][b];
int ans=0;
for(int i=0;i<=A[a];i++)
{
ans=(ans+dfs(a+1,b+i))%mod;
}
B[a][b]=ans;
return ans;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>A[i];
}
cout<<dfs(1,0)<<endl;
return 0;
}