题目:
有N个相同的球,M个不同的盒子,每个盒子最多放K个球
请计算将这N个球全部放入盒子中的方案数模1000007后的结果
n<=5000,m<=5000
输入
6 4 2
输出
10
这道题实际上是维护一个【0~k】的区间和sum,然后滚动一下就好了。
#include<bits/stdc++.h>
using namespace std;
int n,m,k;
const int mod=1000007;
int dp[5010][5010];
int main()
{
cin>>n>>m>>k;
int sum;
for(int i=0;i<=n;i++)
{
dp[i][0]=1;
}
for(int i=1;i<=m;i++)
{
sum=i;
for(int j=1;j<=n;j++)
{
dp[i][j]=sum;
sum=(sum+dp[i-1][j+1])%mod;
if(j>=k)
{
sum=(sum-dp[i-1][j-k]+mod)%mod;
}
}
}
cout<<dp[m][n]<<endl;
return 0;
}