PAT A1103 Integer Factorization (30 分)

题意

题目给出n,k,p，用k个数的p次方的和给出n，若有多个方案，给出和最大的k个数

题目注意点

• 每个数都可以选多次，注意DFS函数的入口
• 先把可能的幂次数存到一个vector里
• 题目要求底数从大到小输出，因此要从最大遍历到最小
• 也需要用vector实时存储当前的方案

题解

思路见算法笔记“一类常见DFS问题的解决方法”
①预处理所有可能出现的底数，存进vector：fac里
②定义DFS函数。递归边界是刚好k个数并且和为n时，此外在超过时也要返回（死胡同）。此处每个数都可以选多次，因此选择了index之后的DFS函数还可以继续传入index
③最终以maxFacSum的值是否为-1作为判断标准输出结果。

#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
int n, k, p, maxFacSum = -1;//maxFacSum记录最大底数之和
vector<int> fac, temp, ans;
//计算power
int power(int x) {
    int ans = 1;
    for (int i = 0; i < p; i++) {
        ans *= x;
    }
    return ans;
}
//预处理fac数组
void init() {
    int i = 0, temp = 0;
    while (temp <= n) {//没超过n时，不断加入fac
        fac.push_back(temp);
        temp = power(++i);
    }
}

void DFS(int index, int nowK, int sum, int facsum) {
    if (nowK == k && sum == n) {
        if (facsum > maxFacSum) {
            ans = temp;//把当前vector复制到结果vector中
            maxFacSum = facsum;
        }
        return;
    }
    if (sum > n || nowK > k) return;
    if (index - 1 >= 0) {
        temp.push_back(index);
        DFS(index, nowK + 1, sum + fac[index], facsum + index);//选
        temp.pop_back();
        DFS(index - 1, nowK, sum, facsum);//不选
    }
}

int main() {
    scanf("%d%d%d", &n, &k, &p);
    init();
    DFS(fac.size() - 1, 0, 0, 0);//从fac的最后一位开始往前搜索
    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;
}