洛谷p1077 摆花

传送门

题目

分析

总共有n种花，最后要选出m盆来摆。
进行逆推，第n种花最多摆an盆，那么最终答案就是sum(f[n-1][m-k])，0<=k<=an
以此类推得到状态转移方程
f[i][j]=sum(f[i-1][j-k])，0<=k<=ai
由于状态转移方程里面没有加常数，很容易想到需要初始化
那么也很容易想到f[0][0]=1，当没有花需要摆的时候有且仅有一种方案

代码

#include <iostream>
using namespace std;
const int N = 110, mod = 1e6+7;
int f[N][N];
int a[N];
int main(){
	int n, m;
	cin>>n>>m;
	for(int i = 1;i <= n;i++) cin>>a[i];
	f[0][0] = 1;
	for(int i = 1;i <= n;i++){
		//这种写法更新的顺序无关紧要，因为没有优化，假如需要优化的话，观察到状态转移方程只与i-1有关，可以变成一维数组，更新的顺序还是令j从m到0
		for(int j = m;j >= 0;j--){
			//要考虑到a[i]可能大于当前容量
			for(int k = 0;k <= min(a[i], j);k++){
				f[i][j] = (f[i][j] + f[i - 1][j - k]) % mod;
			}
		}
	}
	cout<<f[n][m]<<endl;
	return 0;
}

优化

写完状态转移方程，感觉有点眼熟
每次状态只依赖于上一个状态，这不是可以变成一维数组吗
逆序更新保证上一个状态一定还没被更新，也就是用前i-1种花挑出j盆的情况
但是！只是单纯去掉f数组的第一维，居然全红了
为什么呢
当f是二维数组的时候，f[i][j]初值为0，k从0开始迭代相当于先将f[i][j]初始化为f[i-1][j]
但是当f变成一维数组，f[j]的初值不再是0，本身已经是只用前i-1种花选j种的方案数了，就不能从0开始迭代了
修改后的代码如下，成功ac

#include <iostream>
using namespace std;
const int N = 110, mod = 1e6+7;
int f[N];
int a[N];
int main(){
	int n, m;
	cin>>n>>m;
	for(int i = 1;i <= n;i++) cin>>a[i];
	f[0] = 1;
	for(int i = 1;i <= n;i++){
		for(int j = m;j >= 0;j--){
			for(int k = 1;k <= min(a[i], j);k++){
				f[j] = (f[j] + f[j - k]) % mod;
			}
		}
	}
	cout<<f[m]<<endl;
	return 0;
}