题意
有一个正整数 n n n,将其化成 k k k 个正整数的和,问有多少种方案(如果 k k k 个数相同则算一种方案)
思路
看到这个题目的第一思路绝对是爆搜,可是那样会挂的很惨,所以考虑剪枝。
因为如果 k k k 个数相同算一种方案,所以对于每一种相同的方案,我们就取其中单调递增的方案即可。那么有了这个原则,就可以进行剪枝了。
因为方案必须单调递增,所以考虑记录一个 l a s t last last 值表示上一次选择划分的数字,每次枚举此为,就应该从 l a s t last last 开始枚举。每次再记录一个当前选择的和值,我们考虑最坏的情况也就是后面的所有数组都与我们当前这一次一样,那么就应该枚举到当前和加上剩余位数乘当前选择的值。
最后就是找出口了,如果已经选择了 k k k 次,且当前和为 n n n,就代表多了一种方案数,将其计入答案。
代码
#include<bits/stdc++.h>
using namespace std;
int n,k,ans;
void dfs(int last,int sum,int step)
{
if(step==k)
{
if(sum==n) ans++;
return ;
}
for(int i=last;sum+i*(k-step)<=n;i++)
dfs(i,sum+i,step+1);
return ;
}
int main()
{
scanf("%d %d",&n,&k);
dfs(1,0,0);
printf("%d",ans);
return 0;
}