题目 1529: 蓝桥杯算法提高VIP-摆花

题目

小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共m盆。通过调查顾客的喜好，小明列出了顾客最喜欢的n种花，从1到n标号。为了在门口展出更多种花，规定第i种花不能超过ai盆，摆花时同一种花放在一起，且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算，一共有多少种不同的摆花方案。

样例说明
有2种摆花的方案，分别是(1，1，1，2)， (1，1，2，2)。括号里的1和2表示两种花，比如第一个方案是前三个位置摆第一种花，第四个位置摆第二种花。

输入
第一行包含两个正整数n和m，中间用一个空格隔开。
第二行有n个整数，每两个整数之间用一个空格隔开，依次表示a1、a2、……an。

数据规模和约定
对于100%数据，有0< n≤100，0< m≤100，0≤ ai≤100。

输出
输出只有一行，一个整数，表示有多少种方案。注意：因为方案数可能很多，请输出方案数对1000007取模的结果。

样例输入

2 4
3 2

样例输出

2

解题思路

看到这个题目后，应该是可以深度搜索的，但不知时间；为了保险起见，用动态规划去做。本题本质上可以看成是一个物品质量在0-a_i之间自由变化的01背包问题。

假设共有i种花（物品种类数），各种花的数目存储于数组num[i]当中，则要求摆放j盆花（背包容量）时，方案数目的状态转移方程将会是：
$$dp[i][j]=\{\begin{array}{rlr}\sum _{k=0}^{j-1}dp[i-1]& [j-k]+1& (j<=num[i])\\ & 0& (j>num[i])\end{array}$$
综上，按照转移方程来实现即可。

代码

#include<stdio.h>
#define MOD 1000007
long int dp[101][101];
int num[101];

int min(int a, int b){
    return (a>b)?b:a;
}

int main()
{
    int n,m;//摆上一排花，共m盆
    int i,j,k,temp;
    scanf("%d %d",&n,&m);
	for (i=1;i<=n;i++)
	    scanf("%d",&num[i]);
	for (i=1;i<=m;i++)//初始化
	    dp[1][i] = (i>num[1])?0:1;
	for (i=2;i<=n;i++)
	{
	    for (j=1;j<=m;j++)//
	    {
	        for (k=0;k<=num[i];k++)//使用多少朵当前的花
	        {
	            if (k==j)//完全用当前的花填充
	                dp[i][j]++;//将当前的花
	            else if (k<j)
    	            dp[i][j]+=(dp[i-1][j-k])%MOD;
	        }
	        dp[i][j]%=MOD;
	    }
	}
	printf("%ld",dp[n][m]%MOD);
	return 0;
}