摆花 (dp，背包，各种优化)

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

试编程计算，一共有多少种不同的摆花方案。

输入格式
第一行包含两个正整数n和m，中间用一个空格隔开。

第二行有n个整数，每两个整数之间用一个空格隔开，依次表示a1,a2,…,an

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

输入输出样例
输入
2 4
3 2
输出
2
说明/提示
【数据范围】

对于20%数据，有0<n≤8,0<m≤8,0≤a_i≤80<n≤8,0<m≤8,0≤ai≤8；

对于50%数据，有0<n≤20,0<m≤20,0≤a_i≤200<n≤20,0<m≤20,0≤ai≤20；

对于100%数据，有0<n≤100,0<m≤100,0≤ai≤1000<n≤100,0<m≤100,0≤ai≤100。

思路分析：

我一开始想的是dp[i][j]表示已经摆了i盆花，最后摆的花种类为j的方案总数，则最后答案为dp[n][1...m](累加) (这里定义的n为盆数，m为种数)
这样的状态很麻烦。
我们需要：
一层for遍历摆的盆数 i
二层for遍历结尾摆的种类 j
三层for遍历 j 种类摆的盆数 k (k从1开始，不能从0开始，因为已经规定了最后种类为 j )
四层for遍历 该种类在 l (1<=l<=j-1) 后面 （设j=4，可以考虑到前一个种类是1,2,3）

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;

const int mod = 1000007;

int main()
{
	int n, m; //n个m种 
	scanf("%d%d",&m,&n);
	ll dp[102][102];
	ll sum[102][102];
	int a[102];
	memset(dp,0,sizeof(dp));
	memset(sum,0,sizeof(sum));
	for(int i = 1;i <= m;i++) scanf("%d",&a[i]);
	for(int i = 1;i <= m;i++) dp[0][0] = 1, sum[0][i] = sum[0][i-1]+dp[0][i];
	for(int i = 1;i <= n;i++)
	{
		for(int j = 1;j <= m;j++)
		{
			for(int k = 1;k <= a[j] && i-k>=0;k++)
			{
				for(int l = 0;l <= j-1;l++)
				{
					dp[i][j] = (dp[i][j]+dp[i-k][l])%mod;
				}
			}
		}
	}
	ll ans = 0;
	for(int i = 1;i <= m;i++) ans = (ans+dp[n][i])%mod;
	printf("%lld\n",ans);
	return 0;
}

这种时间复杂度很高，但是样例比较水还是AC了。我们试着优化一下。
我们发现第四层循环其实就是dp[i][j] + dp[i-k][1...j-1]，这其实就是前缀和。那么我们可以定义sum[i][j]表示摆了i盆花，最后摆的花种类为[1…j]的的方案总数，则我们最终求得答案就是sum[n][m](n为个数,m为种类)

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
const int mod = 1000007;
int main()
{
	int n, m; //n个m种 
	scanf("%d%d",&m,&n);
	ll dp[102][102]; //dp[i][j]表示摆了i盆花，最后摆的花种类为j的方案总数，则ans=dp[n][1...m]
	ll sum[102][102];	//sum[i][j]表示摆了i盆花，最后摆的花种类为[1,j]的方案总数(前缀和优化) 
	int a[102];
	memset(dp,0,sizeof(dp));
	memset(sum,0,sizeof(sum));
	for(int i = 1;i <= m;i++) scanf("%d",&a[i]);
	for(int i = 0;i <= m;i++) dp[0][0] = 1, sum[0][i] = 1; 
	for(int i = 1;i <= n;i++) //遍历个数 
	{
		for(int j = 1;j <= m;j++) //遍历种类 
		{
			for(int k = 1;k <= a[j] && i-k>=0;k++)	//j种类摆k个 
			{
				dp[i][j] = (dp[i][j]+sum[i-k][j-1])%mod; //j种类摆k个，就由dp[i-k][1...j-1]推来
			}
			sum[i][j] = (sum[i][j-1]+dp[i][j])%mod; //更新前缀和 
		}
	}
	printf("%lld\n",sum[n][m]);
	return 0;
} 

这样优化后N,M取最大也能AC.

这种状态写起来是麻烦的。如果将状态定义为:
dp[i][j]表示在前 n 种花，当前已有 j 盆的方案总数，那么最后结果就是dp[n][m](n为种数，m为个数)
第一层for遍历种类 i
第二层for遍历个数 j ( j 从0开始，因为第一盆可以从2~n起始)
第三层for遍历第 i 种选择放的个数 k ( k 从0开始，设 i = 2, 可以考虑有 [1,3]这种情况)
dp[i][j] = (dp[i][j]+dp[i-1][j-k])%mod

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

int main()
{
	int n, m;	
	const int mod = 1000007;
	scanf("%d%d",&n,&m);
	long long dp[105][105];
	int a[105];
	for(int i = 1;i <= n;i++) scanf("%d",&a[i]);
	memset(dp,0,sizeof(dp));
	dp[0][0] = 1;
	for(int i = 1;i <= n;i++)
	{
		for(int j = 0;j <= m;j++)
		{
			for(int k = 0;k <= a[i] && j-k>=0;k++)
			{
				dp[i][j] = (dp[i][j]+dp[i-1][j-k])%mod;
			}
		}
	}
	printf("%lld\n",dp[n][m]);
	return 0;
}

可以像01背包一样优化一下空间。
我们发现dp[i][j]的更新只和dp[i-1][x](x<=j)层有关系，这和01背包的情况是一样的，因此可以滚动数组优化。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

int main()
{
	int n, m;	
	const int mod = 1000007;
	scanf("%d%d",&n,&m);
	long long dp[105];
	int a[105];
	for(int i = 1;i <= n;i++) scanf("%d",&a[i]);
	memset(dp,0,sizeof(dp));
	dp[0] = 1;
	for(int i = 1;i <= n;i++)
	{
		for(int j = m;j >= 0;j--)
		{
			for(int k = 1;k <= a[i] && j-k>=0;k++)
			{
				dp[j] = (dp[j]+dp[j-k])%mod;
			}
		}
	}
	printf("%lld\n",dp[m]);
	return 0;
}