【PAT甲级题解】1103 Integer Factorization (30分)DFS

Factorization意为“因式分解”，本题考DFS的应用，题目大意是给定N,K,P，将N表示成K个正整数（可以相同，递减排列）的P此方的和，如果有多种方案，那么选择底数和最大的方案，如果还有多种，那么选择底数序列的字典序最大的方案。
因为P>=2而且每次运行的时候都是固定的，所以选择使用一个vector存储所有P次方不超过N的正整数的集合，这个集合中的每一个数就是今后dfs函数中选择“选”或者“不选”的”岔路口”，在该思路的基础上，注意最大子列和的存储与答案的存储，而且为了满足题干”按照子列和最大且按照字典序排列的要求“直接选择从最后一位数开始枚举，这样最终选出的结果就自然满足条件

代码：

#include<cstdio>
#include<vector>
using namespace std;
int k,n,p,maxfacsum = -1;
vector<int>ans,temp,fac;
int power(int x){
    int ans = 1;
    for(int i=0;i<p;i++){
        ans *= x;
    }
    return ans;
}

void init(){
    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;
        }
        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);
    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;
}