一、题目概述
给定正整数N、K、P, 将N划分为K个数的和,且每个数均为某数的P次幂。
二、思路
使用DFS探测所有可能的组合。
递归探测,每次找到一个合适的值,加入数组temp,统计因数的和。当数组temp拥有K个P次幂相加为N的元素,且因数和>Max(当前最大因数和)时,保存temp为备选结果。
void dfs(int margin, int ceil, int sum)
{
if(a.size() < K)
{
for(int i = ceil; i; --i)
if(val[i] <= margin)
{
a.push_back(i);
dfs(margin - val[i], i, sum + i);
a.pop_back();
}
}
else if(!margin && sum > max_sum)
{
ans = a;
max_sum = sum;
}
}
注意:测试点超时dfs中增加和大于N时返回的剪枝判断
三、代码
#include <cstdio>
#include <vector>
#include <cmath>
using namespace std;
int N, K, P, max_sum = 0;
vector<int> ans, a, val;
void dfs(int margin, int ceil, int sum)
{
if(a.size() < K)
{
for(int i = ceil; i; --i)
if(val[i] <= margin)
{
a.push_back(i);
dfs(margin - val[i], i, sum + i);
a.pop_back();
}
}
else if(!margin && sum > max_sum)
{
ans = a;
max_sum = sum;
}
}
int main()
{
scanf("%d %d %d", &N, &K, &P);
for(int i = 0; (int)pow((double)i, P) <= N; ++i)
val.push_back((int)pow((double)i, P));
dfs(N, val.size() - 1, 0);
if(!ans.size())
{
printf("Impossible");
return 0;
}
printf("%d = ", N);
for(int i = 0; i < ans.size(); ++i)
printf("%d^%d%s", ans[i], P, i + 1 == ans.size() ? "":" + ");
}