在网上无意看到的一道题,当时没想出来,后面找了找答案,发现是一个挺典型的动态规划问题,在此做一记录。
问题:有n个肉丸,m个相同的碗,需要把肉丸分到这m个碗里面,求有多少种分配方式。这也相当于是整数拆分问题,即把一个整数n拆分为m个数的和。
思路:起初想着能不能用概率论中的知识直接计算,后面发现算不了,只能通过递推公式以及动态规划计算。
设S(n,m)是n个物体分成m堆的方法数,则有递推公式:
S
(
n
,
m
)
=
S
(
n
−
1
,
m
−
1
)
+
S
(
n
−
m
,
m
)
S(n,m) = S(n-1,m-1) + S(n-m,m)
S(n,m)=S(n−1,m−1)+S(n−m,m)
递推公式解释如下:
n个物体分成m堆,分配结果可分为两种情况:
①至少有一个堆里面只有一个物体
②任意一堆都至少有两个物体
因此,当是情况①时,存在一堆只有一个物体,则把这堆去掉,就变成了n-1个物体分为m-1堆的问题了。
是情况②时,每一堆都至少有两个物体,那么每一堆都分别去掉一个物体,就转变为n-m个物体分为 m堆的问题了。
递推初始值:
S(n,1) = 1 , S(n,n) = 1 ,
n<m时 S(n,m) = 0
C++实现代码如下:
int split_methods(int n,int m)
{
vector<vector<int>>dp(n+1,vector(m+1,0));
for(int i=1;i<n+1;i++)
{
dp[i][1] = 1;
for(int j=2;j<=i && j<m+1;j++)
{
if(j==i){dp[i][j] = 1;continue;}
dp[i][j] = dp[i-1][j-1] + dp[i-j][j];
}
}
return dp[n][m];
}