题目:
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共 �m 盆。通过调查顾客的喜好,小明列出了顾客最喜欢的 �n 种花,从 11 到 �n 标号。为了在门口展出更多种花,规定第 �i 种花不能超过 ��ai 盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
Input
第一行包含两个正整数 n 和 m,中间用一个空格隔开。
第二行有 n 个整数,每两个整数之间用一个空格隔开,依次表示 1,2,⋯,a
1,a2,⋯,an。
Output
一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对 1e6+7取模的结果。
思路:dp,首先定义d[i][j]为第种花,花盆不超过j的摆花方案。
那怎么得到d[i][j]?一般dp都会讨论最后一次选择,这次讨论的是第i种花选择多少,可以是1,2,...a[i],那前i-1种花就是j-1,j-2,...j-a[i];
所以状态方程d[i][j]=d[i][j]+d[i-1][j-k](k<=j&&k<=a[i]);
代码:
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <queue>
#include<algorithm>
#include<cmath>
#include<map>
#include<string.h>
#include<string>
#include<cstring>
using namespace std;
const long long N=1e6;
const long long MAX = 1e6+7;
long long n,m,d[110][120],a[N];
int main()
{
cin >> n >> m;
memset(d, 0, sizeof d);
for (int i = 1; i <= n; i++)
cin >> a[i];
d[0][0] = 1;
for (int i = 1; i <= n; i++)
for (int j = 0; j<=m; j++)//j要充0开始,因为d[n][m]有可能会有前n-1种不选一种,最后第n种花选择m种。
for (int k = 0; k <=a[i]&&k<=j; k++)
d[i][j]=(d[i][j]+d[i-1][j - k])%MAX;
cout << d[n][m] << endl;
return 0;
}