PAT(Advanced Level) 1103 Integer Factorization

Weeping__Angel

已于 2023-01-03 20:45:31 修改

阅读量83

点赞数

文章标签：深度优先算法 c++

于 2023-01-03 17:22:43 首次发布

本文链接：https://blog.csdn.net/ssfdxcj/article/details/128534884

版权

该篇文章介绍了一种使用深度优先搜索（DFS）方法，结合预先计算的pows数组来寻找特定条件下指数和的问题。在DFS过程中进行了剪枝优化，以提高效率。代码使用C++编写，通过递归调用实现搜索，并在满足条件时更新最大和及答案数组。

摘要由CSDN通过智能技术生成

利用 DFS 深度优先搜索
利用 pows 数组存储从 0 开始到 pow(i,d) > n 的 i ^ p 的值。然后 DFS ，其形参为当前正要相加的下标 index（即 pows [ index ] ），当前的总和tempsum，当前相加因子的个数 tempk ，以及当前相加因子的和 sum ( 其与 maxsum 比较，若 sum > maxsum 就更新 maxsum 和 ans数组 ) 。

DFS 还可以进行剪枝：

tempk == k && tempsum != n 时剪枝
当且仅当 tempsum + pows [ index ] <= n时，进行下一层的 DFS ，而不是进入下一层 DFS 后再去判断是否满足条件再返回

#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
int n,k,p,maxsum = -1;
vector<int> pows,ans,tempans;
void pow_list(){
    int temp = 0, index = 1;
    while(temp<=n){
        pows.push_back(temp);
        temp = pow(index,p);
        index++;
    }
}
void dfs(int index,int tempsum,int tempk,int sum){
    if(tempk == k){
        if(tempsum == n && sum > maxsum){
            ans = tempans;
            maxsum = sum;
        }
        return;
    }
    while(index >= 1){
        if(tempsum + pows[index] <= n){
            tempans[tempk] = index;
            dfs(index,tempsum+pows[index],tempk+1,sum+index);
        }
        if(index == 1) return;
        index--;
    }
}

int main(){
    cin >> n >> k >> p;
    pow_list();
    tempans.resize(k);
    dfs(pows.size()-1,0,0,0);
    if(maxsum == -1){
        cout << "Impossible";
        return 0;
    }
    cout << n << " = ";
    for(int i=0;i<ans.size();i++){
        if(i!=0) cout << " + ";
        cout << ans[i] << "^" << p;
    }
}