PAT 1103 Integer Factorization (30 分)

#include <cstdio>
#include <cmath>
#include <vector>
using namespace std;
int n,k,p,maxfacsum=-1;
//fac[i]表示i的p次方
vector<int> fac,ans,temp;
//初始化fac
void init()
{
    int i=1,temp=0;
    while(temp<=n)
    {
        fac.push_back(temp);
        temp=pow(i,p);
        i++;
    }
}
//index表示当前访问fac下标，nowk表示a^b形式的个数，sum表示nowk个a^b的和，facsum表示底数之和
void DFS(int index,int nowk,int sum,int facsum)
{
    //临界条件：如果恰有k个a^b形式，且他们之和为n
    if(sum==n&&nowk==k)
    {
        //更新较优解
        if(facsum>maxfacsum)
        {
            ans=temp;
            maxfacsum=facsum;
        }
        return;
    }
    //超出临界条件，返回
    if(sum>n||nowk>k)
        return;
    //当前fac下标大于等于1，递归
    if(index>=1)
    {
        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;
}