摆花(flower.cpp)
时间限制: 1 Sec
内存限制: 128 MB
提交: 26
解决: 11
[
→][
点击查看最为……的博客][
←]
摆花(flower.cpp)
时间限制: 1 Sec 内存限制: 128 MB提交: 26 解决: 11
[ →][ 点击查看最为……的博客][ ←]
题目描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入
输入共2行。
第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an。
输出
输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
样例输入
2 43 2
样例输出
2
提示
样例说明:
有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]//表示在m号位置放共n种花,此时的方案总数。
=f[n-1][m-1]+f[n-1][m-2]+f[n-1][m-3]+......+f[n-1][0]
因为题目说:花要按顺序摆放,
所以在m号位置放n种花的方案数就只需要看第n-1种花在几号位置停止就可以了。
初始化?我们只需把 只放第一种花的方案数 全部改为1(因为如果只放1号花,只有1种方案)。
你不会以为这样就可以AC了?
状态转移方程式里,我们一直从f[n-1][m-1]加到了f[n-1][0],意图求上一种花在几号位置停下的方案总数,但是,
举个例子:共有3种花,6个位置。
郁金香编号为1,有2朵。
杜鹃花编号为2,有3朵。
百合花编号为3,有2朵。
那么f[3][6]=f[2][5]+f[2][4]+f[2][3]+f[2][2]+f[2][1]+f[2][0]。
问题来了,可以加上f[2][2]?
f[2][2]表示在2号位置放杜鹃,3号位置就要开始摆百合。可是百合只有两朵!根本摆不下!
所以,循环的次数要注意。
详见代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<functional>
using namespace std;
int f[111][111],m,n,a[111];
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=0;i<=a[1];i++) f[1][i]=1;
for(int i=2;i<=n;i++)
for(int j=0;j<=m;j++)
for(int k=0;k<=min(a[i],j);k++)
f[i][j]+=f[i-1][j-k],f[i][j]%=1000007;
printf("%d",f[n][m]);
}