题目大意:
给定正整数NKP,将N表示成K个正整数的p次方的和,如果有多种方案,选择底数和最大的方案,如果还有多种方案,选择底数序列字典序最大的方案。
思路
P不小于2,定义数组fac预处理所有不超过N的n^p
DFS:从fac中选择若干个数(可以重复选)使得他们的和等于N
对当前选择的数的序号记为index,已经选择的数记为nowK,选出的数之和为sum,记录当前选出的底数之和facSum
定义数组and存放最优序列,temp存放当前选中的底数组成的临时序列
为了让字典序大的序列优先被选中,可以让index从大到小递减来遍历,这样总能选中fac中较大的数
停止条件:sum==n&&nowK=k
如果出现sum>N或nowK>K,不可能产生答案,可以直接返回
AC代码:
//PAT_A 1103
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int n, k, p, maxFacSum = -1;//最大底数之和
vector<int> fac, ans, temp;
int power(int x) {
int ans = 1;
for (int i = 0; i < p; i++) {
ans *= x;
}
return ans;
}
void init() {//预处理,i^p没有超过n,把i^p加入fac
int i = 0, temp = 0;
while (temp <= n) {
fac.push_back(temp);
temp = power(++i);
}
}
void dfs(int index, int nowK, int sum, int facSum) {
if (sum == n && nowK == k) {
if (facSum > maxFacSum) {
ans = temp;
maxFacSum = facSum;//更新最大底数之和
}
}
if (sum > n || nowK > k)return;
if (index - 1 >= 0) {
temp.push_back(index);//继续选index
dfs(index, nowK + 1, sum + fac[index], facSum + index);
temp.pop_back();//去掉刚刚加进去的数,再进入不选的分支
dfs(index - 1, nowK, sum, facSum);//不选index
}
}
int main() {
(void)scanf("%d %d %d", &n, &k, &p);
init();
dfs(fac.size() - 1, 0, 0, 0);
if (maxFacSum == -1)printf("Impossible\n");
else {
printf("%d = %d^%d", n, ans[0], p);
for (int i = 1; i < ans.size(); i++) {
printf(" + %d^%d", ans[i], p);
}
}
return 0;
}