题目描述 Description
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入描述 Input Description
输入共2行。
第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an。
输出描述 Output Description
输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
样例输入 Sample Input
2 4
3 2
样例输出 Sample Output
2
数据范围及提示 Data Size & Hint
【输入输出样例说明】
有2种摆花的方案,分别是(1,1,1,2),(1,1,2,2)。括号里的1和2表示两种花,比如第一个方案是前三个位置摆第一种花,第四个位置摆第二种花。
【数据范围】
对于20%数据,有0<n≤8,0<m≤8,0≤ai≤8;
对于50%数据,有0<n≤20,0<m≤20,0≤ai≤20;
对于100%数据,有0<n≤100,0<m≤100,0≤ai≤100。
解析:
f[n][m] 取到第i种花时,还有m盆要取。 f[n][m] = f[n][m] + f[n-1][m-k] 0<=k <= a[i]
m == 0, 说明取完,返回结果 为 1.
m != 0 && i == 0 则说明没有取完,因此该方案结果数为0.
代码:
#include<iostream>
using namespace std;
int num[101];
int n,m;
int f[101][101];
int dfs(int i,int m) // 当前摆到第i种,还剩m盆要摆。
{
if(f[i][m] > 0) return f[i][m];
if(m == 0) return 1;
if(i == 0 && m != 0) return 0;
for(int k = 0; k <= num[i]; k++) // 第i种可以摆k盆
if(m >= k)
f[i][m] = (f[i][m] + dfs(i-1,m-k))%1000007;
return f[i][m];
}
int main()
{
cin >> n >> m;
for(int i = n; i >= 1; i--)
cin >> num[i];
cout<<dfs(n,m);
return 0;
}