P1025 数的划分(C++_dfs_剪枝)

题目描述

将整数n分成k份，且每份不能为空，任意两个方案不相同(不考虑顺序)。

例如：n=7，k=3，下面三种分法被认为是相同的。

1,1,5;
1,5,1;
5,1,1.

问有多少种不同的分法。

输入格式

n,k (6<n≤200，2≤k≤6)

输出格式

1个整数，即不同的分法。

输入输出样例

输入 #1

7 3

输出 #1

4

说明/提示

四种分法为：
1,1,5;
1,2,4;
1,3,3;
2,2,3.

反思

讲真的，乍一看到这个题我的想法非常复杂，甚至想要保存状态，然后每次查找比对状态是否重复，此题的代码是受dalao的启发才出来的。

源码

#include<bits/stdc++.h>
using namespace std;
int ans = 0, n, k;
void dfs(int last, int sum, int num)
{
	if (num == k)
	{
		if(sum == n)
			ans += 1;
		return;
	}
	for (int i = last; sum + i * (k - num) <= n; i++)
		dfs(i, sum + i, num + 1);
}
int main()
{
	cin >> n >> k;
	dfs(1,0,0);
	cout << ans;
	return 0;
}