传送门:
题目描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入输出格式
输入格式:第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an。
输出格式:输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
输入输出样例
2 4 3 2
2
Tip: 这样想,对于前i种花放j盆怎么求呢?用dp[i][j]; dp[i][0] = 1..因为阿如果前i种花放0盆肯定会有一种情况,那就是都不放花
前n种花放j盆和第i种花放1,2,.....j盆有什么联系和区别呢?
区别:前i种类型花比前i-1种类型花 多了一种类型花A
联系:那么花A应该多少盆呢?可以放0盆,1盆,......a[i]盆,恩?这有啥联系的?
前 i种类型花放j盆 : 花A(第i种类型放0盆的情况)dp[i-1][j] (前i-1种花要摆j盆花出来)
花A(第i种类型放1盆的情况)dp[i-1][j-1] (前i-1种花要摆j-1盆花出来)
花A(第i种类型放2盆的情况)dp[i-1][j-2] (前i-1种花要摆j-2盆花出来)
咦,我们发现阿,dp[i][j] = dp[i-1][j] + dp[i-1][j-1] + dp[i-1][j-2] +dp[i-1][j-3]....+dp[i-1][j-k] (j-k>0)
#include <iostream>
#include <cstdio>
const int mmax = 1e3;
const int mod = 1000007;
using namespace std;
int dp[mmax][mmax];
int v[mmax];
int main(){
int n,m;
scanf("%d%d",&n,&m);
for(int i = 1 ; i <= n ; i++) scanf("%d",&v[i]);
for(int i = 0 ; i <= n ;i++) dp[i][0] = 1;
for(int i = 1 ; i <= n ;i++){
for(int j = 1 ; j <= m ;j++){
for(int k = 0 ; k <= v[i] ; k++){ //k代表前i-1种花得放多少盆
if(j-k < 0) break;
dp[i][j] = (dp[i][j] + dp[i-1][j-k])%mod ;
dp[i][j] %= mod ;
}
}
}
cout << dp[n][m] << endl;
}