洛谷 P1077 摆花

目录

• [TOC]
  • 题目
  • 思路
  • 代码

题目

P1077 摆花

思路

( ext{DP})
( ext{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
",f[n][m]%1000007);
	return 0;
}