题目描述
将整数 n 分成 k 份,且每份不能为空,任意两个方案不相同(不考虑顺序)。
例如:n=7,k=3,下面三种分法被认为是相同的。
1,1,51,1,5;
1,5,11,5,1;
5,1,15,1,1.
问有多少种不同的分法。
输入格式
n,k (6<n <=200,2≤k≤6)
输出格式
1 个整数,即不同的分法。
输入输出样例
输入
7 3
输出
4
说明/提示
四种分法为:
1,1,5
1,2,4
1,3,3
2,2,3
40分做法:主要思路:本题很容易想到的方法是枚举1~(n-sum)的k个数,然后去判断当前的和是否等于n,此题还需要考虑到去重,所以可以用一个set集合,把k个数存储进容器再存储进集合中,最后输出集合的个数即可,代码如下:
#include<bits/stdc++.h>
using namespace std;
long long res=0;
set<vector<int> > st;
void fun(int n,int k,int current,vector<int> v)
{
if(k==0)
{
if(current==n)
{
sort(v.begin(),v.end());
st.insert(v);
}
}
else if(k>0)
{
for(int i=1;i<=n-current;i++)
{
v.push_back(i);
fun(n,k-1,current+i,v);
v.pop_back();
}
}
}
int main()
{
int n,k;
cin>>n>>k;
vector<int> v;
fun(n,k,0,v);
int size = st.size();
/*for(set<vector<int> >::iterator it=st.begin();it!=st.end();it++)
{
int length = (*it).size();
for(int i=0;i<length;i++)
{
cout<<(*it)[i]<<" ";
}
cout<<endl;
}*/
cout<<st.size()<<"\n";
return 0;
}
**100分做法:**上面那样做后面三个测试点会超时,仔细思考发现我们再加入集合的时候用到了sort()函数进行排序,使得如果一样的数组都会升序排列,比如5 1 1 、 1 5 1再加入的时候都会变成1 1 5,那么其实可以在for循环里面进行剪枝操作,加上一个限制条件,每次加入的数不得比上一个数小,修改后代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,k,res;
void fun(int last,int sum,int cur)
{
if(cur==k)
{
if(sum==n)
{
res++;
}
return;
}
else if(k>0)
{
//sum+i*(k-cur)<=n 有人判断条件是这个,但是我不是很能想到。
for(int i=last;i<=n-sum;i++)
{
fun(i,sum+i,cur+1);
}
}
}
int main()
{
cin>>n>>k;
fun(1,0,0);
cout<<res<<"\n";
return 0;
}