本题提交时需要注意加上cmath函数,否则编译不了
注意到dfs的一般写法
给定边界判断,不超过边界继续往下走
一般有选择当前节点和不选两种,所以一般有两次dfs。
保留节点采用的是vector存储。
#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
int n,k,p,maxsum=-1;
vector<int> v,ans,temp;
void init()
{
int temp=0,cnt=1;
while(temp<=n)
{
v.push_back(temp);
temp=pow(cnt++,p);
}
}
void DFS(int index,int nowk,int nown,int facsum)
{
if(nown==n&&nowk==k)
{
if(facsum>maxsum)
{
maxsum=facsum;
ans=temp;
}
return;
}
if(nown>n||nowk>k)
return;
if(index>0)
{
temp.push_back(index);
DFS(index,nowk+1,nown+v[index],facsum+index);
temp.pop_back();
DFS(index-1,nowk,nown,facsum);
}
}
int main()
{
cin>>n>>k>>p;
init();
DFS(v.size()-1,0,0,0);
if(maxsum==-1)
printf("Impossible");
else
{
printf("%d = ",n);
for (int i=0;i<ans.size();i++)
{
printf("%d^%d",ans[i],p);
if(i!=ans.size()-1)
printf(" + ");
}
}
}