PAT A.1103 Integer Factorization (DFS)

独自莫凭欄

于 2021-08-10 14:52:24 发布

阅读量110

点赞数

分类专栏：《算法笔记》读书笔记文章标签：动态规划指数表达式质因数深度优先搜索算法优化

本文链接：https://blog.csdn.net/m0_46285887/article/details/119569777

版权

《算法笔记》读书笔记专栏收录该内容

24 篇文章 1 订阅

订阅专栏

在这里插入图片描述
题目链接

#include <bits/stdc++.h>
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(){
    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()
{
    cin >> n >> k >> p;
    Init();
    DFS(fac.size()-1, 0, 0, 0);
    if(maxFacSum == -1){
        cout << "Impossible" << endl;
    }
    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;
}