洛谷P1077题解
题意
中文题惯例了。
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
笺释
想通过这道题好好梳理一下dp方案数的本质。
在用dp求方案数的时候,实际上是在求什么?
是在求组合数。
按照题目要求,有多少种组合情况。
思路的话有两种:
1.对于每一个位置选择放哪一种花,dp[i][j]意为在位置i处放j类的花,有多少种排列方案。我是用刷表法更新的。
for(int i=0;i<=m;i++)
{
for(int j=0;j<=n;j++)
{
if(dp[i][j])
{
for(int k=j+1;k<=n;k++)
{
for(int l=1;l<=a[k];l++)
{
dp[i+l][k]+=dp[i][j];
dp[i+l][k]%=MOD;
}
}
}
}
}
2.dp[i][j]意为用i种花摆j盆的方案数。
for(int i=2;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
for(int k=0;k<=min(a[i],j);k++)
{
dp[i][j]+=dp[i-1][j-k]%1000007;
dp[i][j]%=1000007;
}
}
}
相较于第一种算法,少的一层循环实际上是少在:这种算法所记录的
并不是第i种花摆j盆,而是前i种花摆j盆。
完整代码
#include<bits/stdc++.h>
using namespace std;
int dp[105][105];
int n,m,a[105];
int main(){
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
for(int i=1;i<=n;i++)
{
dp[i][0]=1;
}
for(int i=1;i<=a[1];i++)
{
dp[1][i]=1;
}//dp[1][1]=1,[1][2]=1,[1][3]=1'
for(int i=2;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
for(int k=0;k<=min(a[i],j);k++)
{
dp[i][j]+=dp[i-1][j-k]%1000007;
dp[i][j]%=1000007;
}
}
}
printf("%d",dp[n][m]);
}
#include<bits/stdc++.h>
using namespace std;
#define MAXN 205
#define MOD 1000007
int n,m,a[MAXN],dp[MAXN][MAXN];
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
dp[0][0]=1;
for(int i=0;i<=m;i++)
{
for(int j=0;j<=n;j++)
{
if(dp[i][j])
{
for(int k=j+1;k<=n;k++)
{
for(int l=1;l<=a[k];l++)
{
dp[i+l][k]+=dp[i][j];
dp[i+l][k]%=MOD;
}
}
}
}
}
int ans=0;
for(int i=1;i<=n;i++)
{
ans+=dp[m][i];
ans%=MOD;
}
printf("%d",ans);
}