H106OJ第五次练习:摆花
提示:本文章为课程任务(学习记录)
问题描述
题目:
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入:
第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an
输出:
输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
样例:
输入:
2 4
3 2
输出:
2
思路:
对于此题用到了动态数组,设数组a[]用来存放ai,这道题可以用动态规划来做,建一个数组a,用来存ai,二维数组arr[i][j]表示i种花j盆,arr[i][j] = sum{arr[i-1][j-k]}
k为第i种花摆放的盆数,所以前i-1种花有j-k盆,对前面能摆放的所有可能的方法数求和,即i种花摆放j盆的方法数。比较简单。代码如下:
代码:
#include<iostream>
#include<algorithm>
using namespace std;
int arr[105][105];
int a[105];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1; i<=n; i++)
{
cin>>a[i];
}
arr[0][0]=1;
for(int i=1; i<=n; i++)
{
for(int j=0; j<=m; j++)
{
for(int k=0; k<=min(a[i],j); k++)
{
arr[i][j]=(arr[i][j]+arr[i-1][j-k])%1000007;
}
}
}
cout<<arr[n][m]%1000007<<endl;
return 0;
}