洛谷P1077摆花题解--zhengjun

题面传送门

思路

一看就是一道$dp$

用$f_{i,j}$表示摆到第$i$种花，已经摆了$j$盆花所有的可能性；

初始化：$f_{0,0}=1$

状态转移公式：

$f_{i,j}=\sum _{k=1}^{min(a_i,j)}{f}_{i-1,j-k}$

这里就直接用滚动数组了。

代码

#include<bits/stdc++.h>
#define mod 1000007
using namespace std;
int n,m;
int a[101];
int f[101];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	f[0]=1;
	for(int i=1;i<=n;i++){
		for(int j=m;j>=0;j--){
			for(int k=1;k<=j&&k<=a[i];k++){
				f[j]=(f[j]+f[j-k])%mod;
			}
		}
	}
	printf("%d",f[m]);
	return 0;
}

谢谢–zhengjun