盒子与小球三

题目:
有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;
} 
    原文作者:树人中学李畅
    原文地址: https://blog.csdn.net/qq_43508023/article/details/87867465
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞