PAT (Advanced Level) 1103. Integer Factorization (30)

暴力搜索。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
using namespace std;

int n,k,p,top;
int path[];
int flag=;
int ans_path[],ans_num=,ans_len;

void dfs(int x,int pre,int sum,int num)
{
    if(sum>n) return;
    if(x>k) return;
    if(sum==n)
    {
        if(x!=k) return;

        flag=;
        if(num>ans_num)
        {
            ans_num=num;
            ans_len=x;
            for(int i=;i<x;i++) ans_path[i]=path[i];
        }

        return;
    }
    for(int i=pre;i>=;i--)
    {
        path[x]=i;
        dfs(x+,i,sum+int(pow(i,p)+0.5),num+i);
    }
}
int main()
{
    scanf("%d%d%d",&n,&k,&p);
    top=(int) (pow(1.0*n,1.0/p)+0.5);
    top=min(top,n/k); //加这个剪枝速度还可以提高一些

    dfs(,top,,);
    if(flag==) printf("Impossible\n");
    else
    {
        printf("%d = ",n);

        for(int i=;i<ans_len;i++)
        {
            printf("%d^%d",ans_path[i],p);
            if(i!=ans_len-) printf(" + ");
        }
    }
    return ;
}