题目的意思是给定n,k,p,求是否存在k个正整数,每个数的p次幂相加的结果等于n。
有,输出k个数相加的结果最大的那个,如果有多个,输出序列从大到小排最大的那个(从左往右比较,若
i
<
l
,
a
i
=
b
i
,
a
l
>
b
l
i<l,ai=bi,a_l>b_l
i<l,ai=bi,al>bl,则序列a大于序列b)。
这一题一开始毫无思路,如果暴力肯定会超时。随后查了别人的讲解,有人就简单提到深度优先搜索,还是想不出来。看了一下柳神的讲解,挺详细,原来是先把从i=0开始p次方小于等于n的数保存下来,然后用深度优先搜索,每次把值加上去,判断是否大于n,不是就接着搜索。
这一题一开始写好,我每次搜索都是从头开始搜索,第二个测试点自己测的时候要很久才出来,比较了一下与别人的代码,发现可以每次可以从上一个元素的位置开始遍历,因为上一个元素的前一个元素前面已经遍历过,所以这次不用。而且求的是从大到小最大序列,就不用重复计算。
遍历的时候最好从大到小遍历,因为这样,就不用比较相等的情况,大的序列肯定在前面产生。
今天才发现深度优先搜索还可以这样用,以前只认为用在图中,也算是让我的思维打开了一点,加深了一下对深度优先搜索的理解。
(用时:1:41:22.47)
附上别人的讲解:
https://blog.csdn.net/liuchuo/article/details/52493390
#include <bits/stdc++.h>
using namespace std;
vector<int> numVec;
int maxNumSum = -1;
int vecSize = 0;
int maxNum[401];
void dfs(int numSum,int currenSum,int k,int addCount,int n,int num[],int index)
{
if(k==addCount) {
if(currenSum==n) {
if(maxNumSum < numSum) {
maxNumSum = numSum;
for(int i=0; i<k; i++) {
maxNum[i] = num[i];
}
}
}
return;
}
for(int i=index; i>=1; i--) {
if(currenSum + numVec[i] <= n) {
num[addCount] = i;
dfs(numSum + i, currenSum + numVec[i], k, addCount+1,n,num,i);
}
}
}
int main()
{
int n,k,p;
scanf("%d%d%d",&n,&k,&p);
int i=0;
int iPow = pow(i,p);
while(iPow <=n) {
i++;
numVec.push_back(iPow);
iPow =1;
for(int j=0; j<p; j++)
iPow *= i;
}
vecSize = numVec.size();
int num[k];
dfs(0,0,k,0,n,num,vecSize-1);
if(maxNumSum == -1) {
printf("Impossible\n");
} else {
printf("%d =",n);
for( i=0; i<k; i++) {
if(i==0) {
printf(" %d^%d",maxNum[i],p);
} else {
printf(" + %d^%d",maxNum[i],p);
}
}
printf("\n");
}
return 0;
}