洛谷 P1077 摆花

[TOC]

题目

P1077 摆花

思路

\(\text{DP}\)
$\text{f[i][j]}$表示前$i$种花摆了$j$盆的方案数。
设$x$（$x$的范围为$[0,a_i]$）为第$i$种花摆了多少盆，\(k\)（$k$的范围为$[0,m-x]$）为前面$i-1$种花摆了多少盆。枚举$x$，\(k\)。
所以$f[i][x+k]=f[i][x+k]+f[i-1][k]$

代码

#include<bits/stdc++.h>
using namespace std;
int n,m;
int f[101][101];
inline int read(){
	int x=0;bool f=0;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=!f;c=getchar();}
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return f?-x:x;
}

int main(){
	n=read(),m=read();
	for(int i=0;i<=n;++i){
		f[i][0]=1;
	}
	for(int i=1,num;i<=n;++i){
		num=read();
		for(int j=0;j<=num;++j){
			for(int k=0;k<=m-j;++k){
				if(j==0&&k==0) continue;
				f[i][j+k]+=f[i-1][k];
				f[i][j+k]%=1000007;
			}
		}
	}
	printf("%d\n",f[n][m]%1000007);
	return 0;
}