题目
在花店的门口摆上一排花,共mmm盆。有nnn种花,从111到nnn标号。规定第iii种花不能超过aia_iai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。问一共有多少种不同的摆花方案。
分析
其实这长得很像01背包,但是状态转移方程
f[i][j]f[i][j]f[i][j]表示前i种花摆出j种的方案数
f[i][j]=∑k=0min{a[i],j}f[i−1][k]f[i][j]=\sum_{k=0}^{min\{a[i],j\}}f[i-1][k]f[i][j]=k=0∑min{a[i],j}f[i−1][k]
滚动数组代码
#include <cstdio>
#define rr register
using namespace std;
int f[2][101],n,m;
inline signed iut(){
rr int ans=0; rr char c=getchar();
while (c<48||c>57) c=getchar();
while (c>47&&c<58) ans=(ans<<3)+(ans<<1)+c-48,c=getchar();
return ans;
}
signed main(){
n=iut(),m=iut(),f[0][0]=1;
for (rr int i=1,x;i<=n;++i){
x=iut();
for (rr int j=0;j<=m;++j){
f[i&1][j]=0; rr int t=(x<j)?x:j;
for (rr int k=0;k<=t;++k)
f[i&1][j]=(f[i&1][j]+f[!(i&1)][j-k])%1000007;
}
}
return !printf("%d",f[n&1][m]);
}