第一段代码(三个TLE)简单的用了dfs,保存下来所有序列,然后去重。。。第二段代码参考大佬的剪枝(可以AC)只能说想不到。。。
#include<bits/stdc++.h>
using namespace std;
int n,k;
vector<vector<int> > ppp;
map<vector<int>,bool> mp;
vector<int> temp;
void dfs(int sum,int num,int lastid){
if(num==k&&sum==n){
ppp.push_back(temp);
return;
}
if(num>k||sum>n) return;
for(int i=lastid;i<n;i++){
temp.push_back(i);
dfs(sum+i,num+1,lastid);
temp.pop_back();
}
}
int main()
{
cin>>n>>k;
dfs(0,0,1);
int cnt=0;
for(int i=0;i<ppp.size();i++){
sort(ppp[i].begin(),ppp[i].end());
if(mp[ppp[i]]==false){
mp[ppp[i]]=true;cnt++;
/*for(int j=0;j<ppp[i].size();j++){
cout<<ppp[i][j]<<' ';
}
cout<<endl;*/
}
}
cout<<cnt;
return 0;
}
#include<bits/stdc++.h>
using namespace std;
int n,k;
int cnt=0;
void dfs(int sum,int num,int lastid){
if(num==k){
if(sum==n)
cnt++;
return;
}
//if(num>k||sum>n) return;
for(int i=lastid;sum+i*(k-num)<=n;i++){
dfs(sum+i,num+1,i);
}
}
int main()
{
cin>>n>>k;
dfs(0,0,1);
cout<<cnt;
return 0;
}